Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Use int32_t for kThreadGroupSize in cta_per_row #1338

Closed
wants to merge 2 commits into from

Conversation

sryap
Copy link
Contributor

@sryap sryap commented Sep 19, 2022

Summary:
Use int32_t for kThreadGroupSize instead of size_t in TBE bwd
warp_per_row. Using size_t causes a slowdown in some cases.

Differential Revision: D39637900

@netlify
Copy link

netlify bot commented Sep 19, 2022

Deploy Preview for eclectic-stroopwafel-199537 canceled.

Name Link
🔨 Latest commit 220c79f
🔍 Latest deploy log https://app.netlify.com/sites/eclectic-stroopwafel-199537/deploys/6329565eb26d4c0008ea2e94

@facebook-github-bot
Copy link
Contributor

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

Sarunya Pumma and others added 2 commits September 19, 2022 22:56
Differential Revision: D39590263

fbshipit-source-id: e69da30cb5f89d70a796a076f0d80100233fc100
Summary:
Pull Request resolved: pytorch#1338

Use int32_t for kThreadGroupSize instead of size_t in TBE bwd
warp_per_row.  Using size_t causes a slowdown in some cases.

Differential Revision: D39637900

fbshipit-source-id: 3896c988aef660953629fde1b4feadf2247009ca
@facebook-github-bot
Copy link
Contributor

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

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