commit d11603de3814bb337b3875dec32ac25a2e6c6486
author Mikhail Zolotukhin <mvz@fb.com> Thu Sep 03 11:23:17 2020 -0700
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> Thu Sep 03 11:29:35 2020 -0700
tree 962361dec0cba61b41e69fcc85d3eebadac2fce5
parent b10c527a1fbe56273d096c97f6f55bb43d107776

[TensorExpr] Benchmarks: set number of profiling runs to 2 for PE. (#44112)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/44112

Test Plan: Imported from OSS

Reviewed By: bertmaher

Differential Revision: D23500904

Pulled By: ZolotukhinM

fbshipit-source-id: d0dd54752b7ea5ae11f33e865c96d2d61e98d573

benchmarks/fastrnns/fuser.py

diff --git a/benchmarks/fastrnns/fuser.py b/benchmarks/fastrnns/fuser.py
index 3ed1699..620c19a 100644
--- a/benchmarks/fastrnns/fuser.py
+++ b/benchmarks/fastrnns/fuser.py
@@ -6,6 +6,7 @@
         torch._C._jit_set_profiling_executor(True)
         torch._C._jit_set_profiling_mode(True)
         torch._C._jit_set_bailout_depth(20)
+        torch._C._jit_set_num_profiled_runs(2)
         torch._C._jit_override_can_fuse_on_cpu(False)
         torch._C._jit_override_can_fuse_on_gpu(True)
         torch._C._jit_set_texpr_fuser_enabled(True)
@@ -26,6 +27,7 @@
         torch._C._jit_set_profiling_executor(True)
         torch._C._jit_set_profiling_mode(True)
         torch._C._jit_set_bailout_depth(20)
+        torch._C._jit_set_num_profiled_runs(2)
     elif executor_name == 'simple':
         torch._C._jit_set_profiling_executor(True)
         torch._C._jit_set_profiling_mode(False)