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

EmbeddingLayer CUDA performance #7180

Closed
raver119 opened this issue Feb 17, 2019 · 2 comments

Comments

Projects
None yet
1 participant
@raver119
Copy link
Contributor

commented Feb 17, 2019

EmbeddingLayer uses ScatterUpdate op, which is CPU-only atm.
If we do beta4 release before CUDA support for new ops gets merged, we should provide temporary solution for it, because performance loss is just ridiculous in this use case.

@raver119

This comment has been minimized.

Copy link
Contributor Author

commented Feb 17, 2019

@lock

This comment has been minimized.

Copy link

commented Mar 23, 2019

This thread has been automatically locked since there has not been any recent activity after it was closed. Please open a new issue for related bugs.

@lock lock bot locked and limited conversation to collaborators Mar 23, 2019

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
You can’t perform that action at this time.