pytorch/benchmarks/static_runtime
Scott Wolchok 98d1529474 [PyTorch] fix mixed int32/int64 indices/offsets for embedding_bag_out (#120752)
This was an oversight in D27482738 (#55189) -- it only patched the regular embedding_bag operator, but static runtime uses the out variant.

Differential Revision: [D54285460](https://our.internmc.facebook.com/intern/diff/D54285460/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/120752
Approved by: https://github.com/houseroad
2024-02-28 20:13:30 +00:00
..
CMakeLists.txt
deep_wide_pt.cc
deep_wide_pt.h
deep_wide_pt_bench.cc
test_cpu_fusion.cc
test_generated_ops.cc
test_static_module.cc
test_static_runtime.cc
test_utils.cc
test_utils.h