enable NVFuser by default

davidberard98 · davidberard98 · commit 7ea37f556364 · 2022-05-10T18:03:05.000-07:00
Enable NVFuser in OSS. Tests are passing, and we've also run tests in [torchvision](pytorch/vision#5959) and [torchaudio](pytorch/audio#2372) Retry of #76006, because that PR had GH1/ghstack issues. [ghstack-poisoned]
diff --git a/torch/csrc/jit/codegen/cuda/interface.cpp b/torch/csrc/jit/codegen/cuda/interface.cpp
@@ -90,9 +90,12 @@ class NVFuserEnabler {
     if (getCachedFuserEnabledEnvVar().has_value()) {
       return *getCachedFuserEnabledEnvVar();
     }
-    // 3. default value (if you switch this to true, make sure
-    //    to check nvfuserCanBeEnabled())
+    // 3. default value
+#ifdef FBCODE_CAFFE2
     return false;
+#else
+    return nvfuserCanBeEnabled();
+#endif
   }
 
  public:

Original file line number	Diff line number	Diff line change
`@@ -90,9 +90,12 @@ class NVFuserEnabler {`
`90`	`90`	`if (getCachedFuserEnabledEnvVar().has_value()) {`
`91`	`91`	`return *getCachedFuserEnabledEnvVar();`
`92`	`92`	`}`
`93`		`- // 3. default value (if you switch this to true, make sure`
`94`		`- // to check nvfuserCanBeEnabled())`
	`93`	`+ // 3. default value`
	`94`	`+#ifdef FBCODE_CAFFE2`
`95`	`95`	`return false;`
	`96`	`+#else`
	`97`	`+ return nvfuserCanBeEnabled();`
	`98`	`+#endif`
`96`	`99`	`}`
`97`	`100`
`98`	`101`	`public:`