Skip to content

[ROCm] Enable test_nn embedding tests and use correct warp size in Embedding.cu#13046

Closed
iotamudelta wants to merge 1 commit intopytorch:masterfrom
ROCm:embedding_20181024
Closed

[ROCm] Enable test_nn embedding tests and use correct warp size in Embedding.cu#13046
iotamudelta wants to merge 1 commit intopytorch:masterfrom
ROCm:embedding_20181024

Conversation

@iotamudelta
Copy link
Copy Markdown
Contributor

  • Enable test_nn embedding tests and use correct warp size in Embedding.cu
  • Fix embedding_backward_feature_kernel kernel for HIP

For attention: @bddppq @ezyang

….cu (ROCm 285)

* Enable test_nn embedding tests and use correct warp size in Embedding.cu

* Fix embedding_backward_feature_kernel kernel for HIP
@iotamudelta
Copy link
Copy Markdown
Contributor Author

@iotamudelta iotamudelta changed the title Enable test_nn embedding tests and use correct warp size in Embedding.cu [ROCm] Enable test_nn embedding tests and use correct warp size in Embedding.cu Oct 24, 2018
Copy link
Copy Markdown
Contributor

@facebook-github-bot facebook-github-bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ezyang has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

@bddppq bddppq added the module: rocm AMD GPU support for Pytorch label Oct 24, 2018
Copy link
Copy Markdown
Contributor

@facebook-github-bot facebook-github-bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

bddppq is landing this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

Copy link
Copy Markdown
Contributor

@facebook-github-bot facebook-github-bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

bddppq is landing this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

zdevito pushed a commit to zdevito/ATen that referenced this pull request Oct 25, 2018
….cu (#13046)

Summary:
* Enable test_nn embedding tests and use correct warp size in Embedding.cu
* Fix embedding_backward_feature_kernel kernel for HIP

For attention: bddppq ezyang
Pull Request resolved: pytorch/pytorch#13046

Differential Revision: D10560721

Pulled By: bddppq

fbshipit-source-id: e6c3cbeb980a34ff52a92dba8bde745a2e03f2fd
laurentdupin pushed a commit to laurentdupin/pytorch that referenced this pull request Apr 24, 2026
….cu (pytorch#13046)

Summary:
* Enable test_nn embedding tests and use correct warp size in Embedding.cu
* Fix embedding_backward_feature_kernel kernel for HIP

For attention: bddppq ezyang
Pull Request resolved: pytorch#13046

Differential Revision: D10560721

Pulled By: bddppq

fbshipit-source-id: e6c3cbeb980a34ff52a92dba8bde745a2e03f2fd
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

module: rocm AMD GPU support for Pytorch open source

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants