-
Notifications
You must be signed in to change notification settings - Fork 486
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
Add support for int64_t indices and offsets in TBE inference [7B/N] #3189
base: main
Are you sure you want to change the base?
Commits on Sep 30, 2024
-
Add support for int64_t indices in TBE inference [1/N]
Summary: - Add support for int64_t indices in TBE inference [1/N] Differential Revision: D61813383
Configuration menu - View commit details
-
Copy full SHA for 55431bd - Browse repository at this point
Copy the full SHA 55431bdView commit details -
Add support for int64_t indices in TBE inference [2/N]
Summary: - Add support for int64_t indices in TBE inference [2/N] - Convert `pruned_array_lookup_cuda` to use index_t Differential Revision: D62271409
Configuration menu - View commit details
-
Copy full SHA for b57f0ee - Browse repository at this point
Copy the full SHA b57f0eeView commit details -
Add support for int64_t indices and offsets in TBE inference [3/N]
Summary: - Convert `pruned_hashmap_lookup_cuda` to use index_t Differential Revision: D62277673
Configuration menu - View commit details
-
Copy full SHA for 848ba70 - Browse repository at this point
Copy the full SHA 848ba70View commit details -
Add support for int64_t indices and offsets in TBE inference [4/N]
Summary: - Convert `pruned_array_lookup_cpu` to use `index_t` Differential Revision: D62470736
Configuration menu - View commit details
-
Copy full SHA for cfcaa56 - Browse repository at this point
Copy the full SHA cfcaa56View commit details -
Add support for int64_t indices and offsets in TBE inference [5/N]
Summary: - Convert `pruned_hashmap_lookup_cpu` to use `index_t` Differential Revision: D62472965
Configuration menu - View commit details
-
Copy full SHA for 61e73ea - Browse repository at this point
Copy the full SHA 61e73eaView commit details -
Add support for int64_t indices and offsets in TBE inference [6/N]
Summary: - Convert `PrunedMapCPU::lookup` to use `index_t` Differential Revision: D62602764
Configuration menu - View commit details
-
Copy full SHA for 4a8b5d0 - Browse repository at this point
Copy the full SHA 4a8b5d0View commit details
Commits on Oct 1, 2024
-
Add support for int64_t indices and offsets in TBE inference [7B/N] (p…
…ytorch#3189) Summary: X-link: facebookresearch/FBGEMM#284 Pull Request resolved: pytorch#3189 - Fix `index_remapping` in `pruned_array_lookup` to be `int64_t` always, since it is set up before we are provided `indices` and `offsets` Differential Revision: D63567553
Configuration menu - View commit details
-
Copy full SHA for 5984c26 - Browse repository at this point
Copy the full SHA 5984c26View commit details