enable PE everywhere but mobile (#47001)
Summary:
enable PE everywhere but mobile
Pull Request resolved: https://github.com/pytorch/pytorch/pull/47001
Reviewed By: eellison
Differential Revision: D24596252
Pulled By: Krovatkin
fbshipit-source-id: 3e3093a43287e1ff838cb03ec0e53c11c82c8dd2
diff --git a/torch/csrc/jit/runtime/profiling_graph_executor_impl.cpp b/torch/csrc/jit/runtime/profiling_graph_executor_impl.cpp
index 3d7e8bb..d62697c 100644
--- a/torch/csrc/jit/runtime/profiling_graph_executor_impl.cpp
+++ b/torch/csrc/jit/runtime/profiling_graph_executor_impl.cpp
@@ -40,8 +40,7 @@
namespace torch {
namespace jit {
-// TODO: keep the else clause for trial runs
-#if defined(FBCODE_CAFFE2) || defined(C10_MOBILE)
+#if defined(C10_MOBILE)
static std::atomic<bool> executor_mode{true};
static std::atomic<bool> profiling_mode{false};
#else