diff --git a/benchmarks/fastrnns/fuser.py b/benchmarks/fastrnns/fuser.py index 620c19a13cf..3c1b87f010e 100644 --- a/benchmarks/fastrnns/fuser.py +++ b/benchmarks/fastrnns/fuser.py @@ -1,7 +1,7 @@ import torch def set_fuser(fuser_name, executor_name): - assert fuser_name in ['te', 'old', 'none'] + assert fuser_name in ['te', 'old', 'none', 'default'] if fuser_name == 'te': torch._C._jit_set_profiling_executor(True) torch._C._jit_set_profiling_mode(True) @@ -21,6 +21,8 @@ def set_fuser(fuser_name, executor_name): torch._C._jit_override_can_fuse_on_gpu(False) torch._C._jit_override_can_fuse_on_cpu(False) torch._C._jit_set_texpr_fuser_enabled(False) + elif fuser_name == 'default': + pass # --executor overrides settings of --fuser if executor_name == 'profiling': @@ -34,3 +36,5 @@ def set_fuser(fuser_name, executor_name): elif executor_name == 'legacy': torch._C._jit_set_profiling_executor(False) torch._C._jit_set_profiling_mode(False) + elif executor_name == 'default': + pass