Fix the te fuser setup (#963)

Summary:
Correctly enable `te` fuser environment when fuser is specified as `te`.
Fixes #956

Pull Request resolved: #963

Reviewed By: davidberard98

Differential Revision: D37218257

Pulled By: xuzhao9

fbshipit-source-id: 07019cd3fd302213eb8c21245abaa9fcfbb3b6ae

conftest.py

@@ -21,9 +21,11 @@ def pytest_addoption(parser):
 
 def set_fuser(fuser):
  if fuser == "te":
- # use default fuser (TE) and
- # profiling executor settings
- pass
+ torch._C._jit_set_profiling_executor(True)
+ torch._C._jit_set_profiling_mode(True)
+ torch._C._jit_override_can_fuse_on_cpu(True)
+ torch._C._jit_override_can_fuse_on_gpu(True)
+ torch._C._jit_set_texpr_fuser_enabled(True)
  elif fuser == "old":
  torch._C._jit_set_profiling_executor(False)
  torch._C._jit_set_profiling_mode(False)