pytorch
74a3ba5e
- [cpp] remove checks from embedding bag impl
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
[cpp] remove checks from embedding bag impl These checks incur an H2D sync on every embedding bag forward. Also, the equivalent python code for embedding_bag does not have them. Kill!
References
embeddingbag
Author
suo
Parents
b399007a
Loading