pytorch/benchmarks/static_runtime
Edward Z. Yang 5747a51657 Fix flaky StaticRuntime.Nonzero test (#94418)
If the operator produces a zero size tensor, the memory
may be equal to the original.  With nonzero, we would sometimes
get unlucky and everything was zero.

See failing tests at https://hud.pytorch.org/failure/%5B%20%20FAILED%20%20%5D%20StaticRuntime.Nonzero

Arguably we should also fix the seeding but it was less obvious
to me where to do that.

Signed-off-by: Edward Z. Yang <ezyang@meta.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/94418
Approved by: https://github.com/albanD
2023-02-16 21:25:15 +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 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