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 [Static Runtime] Add auto-generated out variant dispatchers (#72603) 2022-02-28 08:39:10 +00:00
deep_wide_pt_bench.cc Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032) 2022-07-26 01:20:44 +00:00
deep_wide_pt.cc [NOOP][clangformat][codemod] Enable CLANGFORMAT for some folders in caffe2/* (#67746) 2021-11-03 12:23:14 -07:00
deep_wide_pt.h [SR] Fix broken unit test build (#76111) 2022-04-20 18:36:31 +00:00
test_cpu_fusion.cc [Static Runtime] Add test that runs NNC fused kernels in parallel (#73256) 2022-02-28 17:44:45 +00:00
test_generated_ops.cc Revive static_runtime_benchmark build and test (#87660) 2022-11-08 08:32:45 +00:00
test_static_module.cc fix some typos (#106018) 2023-07-26 18:14:44 +00:00
test_static_runtime.cc [PyTorch] fix mixed int32/int64 indices/offsets for embedding_bag_out (#120752) 2024-02-28 20:13:30 +00:00
test_utils.cc Fix flaky StaticRuntime.Nonzero test (#94418) 2023-02-16 21:25:15 +00:00
test_utils.h [BE] Tweak Meta copyright headers (#90805) 2022-12-14 20:30:31 +00:00