Skip to content

Use less thread blocks for find_uncached kernel#3101

Closed
levythu wants to merge 1 commit into
pytorch:mainfrom
levythu:export-D62401630
Closed

Use less thread blocks for find_uncached kernel#3101
levythu wants to merge 1 commit into
pytorch:mainfrom
levythu:export-D62401630

Conversation

@levythu

@levythu levythu commented Sep 9, 2024

Copy link
Copy Markdown
Contributor

Summary:
Find uncached takes quite long time and this often overalap with the main stream in prefetch mode. We do want it to be overlapped with other kernels so it has minimal effect on the main stream.

This will likely make cache prefetch taking slightly longer (not significantly as prefetch spent longest time in data transfer), but will make overlapping main stream kernels faster.

Differential Revision: D62401630

@netlify

netlify Bot commented Sep 9, 2024

Copy link
Copy Markdown

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
🔨 Latest commit fb6eb24
🔍 Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/66df65041601bc00082176ed
😎 Deploy Preview https://deploy-preview-3101--pytorch-fbgemm-docs.netlify.app
📱 Preview on mobile
Toggle QR Code...

QR Code

Use your smartphone camera to open QR code link.

To edit notification comments on pull requests, go to your Netlify site configuration.

@facebook-github-bot

Copy link
Copy Markdown
Contributor

This pull request was exported from Phabricator. Differential Revision: D62401630

Summary:
Pull Request resolved: pytorch#3101

X-link: facebookresearch/FBGEMM#190

Find uncached takes quite long time and this often overalap with the main stream in prefetch mode. We do want it to be overlapped with other kernels so it has minimal effect on the main stream.

This will likely make cache prefetch taking slightly longer (not significantly as prefetch spent longest time in data transfer), but will make overlapping main stream kernels faster.

Differential Revision: D62401630
@facebook-github-bot

Copy link
Copy Markdown
Contributor

This pull request was exported from Phabricator. Differential Revision: D62401630

@facebook-github-bot

Copy link
Copy Markdown
Contributor

This pull request has been merged in 20cb987.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants