Enable GPU fusion with TEs

pytorch · Sep 3, 2020 · d42743b · d42743b
1 parent 17338f7
commit d42743b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/conftest.py b/conftest.py
@@ -16,7 +16,7 @@ def set_fuser(fuser):
  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(False)
+ torch._C._jit_override_can_fuse_on_gpu(True)
  torch._C._jit_set_texpr_fuser_enabled(True)
 
 def pytest_configure(config):