forked from pytorch/FBGEMM
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
add option to use int64_t offsets/lengths in embedding operators (pyt…
…orch#350) Summary: Pull Request resolved: pytorch#350 PyTorch passes int64_t offsets. Inside the JIT'ed code, we actually cast int64 offsets/lengths to int32 but in practice this shouldn't be a problem unless we have more than 2B indices. Other changes: Move inst_set_t template parameter from generate function to class template parameter to follow the changes in other code generators in fbgemm Reviewed By: jianyuh Differential Revision: D20959297 fbshipit-source-id: c567ed979e8a35eefb04d3a4ce57c8afa13b066d
- Loading branch information
1 parent
20b6cf1
commit 3db8b84
Showing
13 changed files
with
1,948 additions
and
1,427 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.