commit a55ef69e6803df87ae69e0a669e48bf4989ce42a
author Elias Ellison <eellison@devfair044.h1.fair> Tue Feb 01 11:00:10 2022 -0800
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Tue Feb 01 19:07:02 2022 +0000
tree d0d9c4d40f025b717694f6aedd2d5d9ed2a4d23e
parent b44f724aef149124e8871becc7a00fc2dea894c2

update default fusion strategy (#72038)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/72038

Test Plan: Imported from OSS

Reviewed By: navahgar

Differential Revision: D33864653

Pulled By: eellison

fbshipit-source-id: 9f0d7fa5f72a901566fae937668d3a6ede2c4b03
(cherry picked from commit aeee43e8d92255d2926a7ce0f540ffb46a681d6a)

torch/csrc/jit/runtime/profiling_graph_executor_impl.cpp

diff --git a/torch/csrc/jit/runtime/profiling_graph_executor_impl.cpp b/torch/csrc/jit/runtime/profiling_graph_executor_impl.cpp
index 6c87b80..3b7752e 100644
--- a/torch/csrc/jit/runtime/profiling_graph_executor_impl.cpp
+++ b/torch/csrc/jit/runtime/profiling_graph_executor_impl.cpp
@@ -69,7 +69,13 @@
 #endif
 
 static std::mutex fusion_strategy_lock;
+
+// TODO remove ifdef
+#ifdef FBCODE_CAFFE2
 static FusionStrategy fusion_strategy = {{FusionBehavior::STATIC, 20}};
+#else
+static FusionStrategy fusion_strategy = {{FusionBehavior::STATIC, 2}, {FusionBehavior::DYNAMIC, 10}};
+#endif
 
 FusionStrategy getFusionStrategy() {
   std::lock_guard<std::mutex> guard(fusion_strategy_lock);