pytorch
74a3ba5e - [cpp] remove checks from embedding bag impl

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
Author
suo suo
Parents
Loading