Skip TorchScript backend for OVRSource as well (#75138)

malfet · pytorchmergebot · commit 8b7e2bf7a656 · 2022-04-02T04:01:57.000Z
Summary: Pull Request resolved: #75138 (Note: this ignores all push blocking failures!) Test Plan: CI Reviewed By: wconstab Differential Revision: D35331263 fbshipit-source-id: e426c4017359c9f98188c0df5226775be7b1f700 (cherry picked from commit bf1768f)
diff --git a/torch/csrc/lazy/python/init.cpp b/torch/csrc/lazy/python/init.cpp
@@ -11,9 +11,9 @@
 #include <torch/csrc/lazy/core/ir_dump_util.h>
 #include <torch/csrc/lazy/core/internal_ops/ltc_ops.h>
 #include <torch/csrc/lazy/ts_backend/ops/device_data.h>
-#ifndef FBCODE_CAFFE2
+#if !(defined(FBCODE_CAFFE2) || defined(OVRSOURCE))
 #include <torch/csrc/lazy/ts_backend/ts_backend_impl.h>
-#endif // FBCODE_CAFFE2
+#endif // FBCODE_CAFFE2 || OVRSOURCE
 #include <string>
 #include <vector>
 
@@ -161,10 +161,10 @@ void initLazyBindings(PyObject* module){
   lazy_ts_backend.def(
     "_init",
     []() {
-#ifndef FBCODE_CAFFE2
+#if !(defined(FBCODE_CAFFE2) || defined(OVRSOURCE))
       torch::lazy::InitTorchScriptBackend();
 #else
-      TORCH_CHECK(false, "TorchScript backend not yet supported in FBCODE builds");
+      TORCH_CHECK(false, "TorchScript backend not yet supported in FBCODE/OVRSOURCE builds");
 #endif // FBCODE_CAFFE2
     });
 
