Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Enable half for CUDA dense EmbeddingBag backward.#19293

Closed
zou3519 wants to merge 1 commit into
pytorch:masterfrom
zou3519:eb-half
Closed

Enable half for CUDA dense EmbeddingBag backward.#19293
zou3519 wants to merge 1 commit into
pytorch:masterfrom
zou3519:eb-half

Conversation

@zou3519
Copy link
Copy Markdown
Contributor

@zou3519 zou3519 commented Apr 15, 2019

I audited the relevant kernel and saw it accumulates a good deal into float
so it should be fine.

Test Plan:

  • New tests

@zou3519 zou3519 requested a review from cpuhrsch April 15, 2019 22:20
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.

@zou3519 has imported 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.

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

I audited the relevant kernel and saw it accumulates a good deal into float
so it should be fine.

Test Plan:
- New tests
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.

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

@facebook-github-bot
Copy link
Copy Markdown
Contributor

@zou3519 merged this pull request in 3b29cba.

zdevito pushed a commit to zdevito/ATen that referenced this pull request Apr 16, 2019
Summary:
I audited the relevant kernel and saw it accumulates a good deal into float
so it should be fine.
Pull Request resolved: pytorch/pytorch#19293

Differential Revision: D14942274

Pulled By: zou3519

fbshipit-source-id: 36996ba0fbb29fbfb12b27bfe9c0ad1eb012ba3c
zhangguanheng66 pushed a commit to zhangguanheng66/pytorch that referenced this pull request May 6, 2019
Summary:
I audited the relevant kernel and saw it accumulates a good deal into float
so it should be fine.
Pull Request resolved: pytorch#19293

Differential Revision: D14942274

Pulled By: zou3519

fbshipit-source-id: 36996ba0fbb29fbfb12b27bfe9c0ad1eb012ba3c
laurentdupin pushed a commit to laurentdupin/pytorch that referenced this pull request Apr 24, 2026
Summary:
I audited the relevant kernel and saw it accumulates a good deal into float
so it should be fine.
Pull Request resolved: pytorch#19293

Differential Revision: D14942274

Pulled By: zou3519

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants