Skip to content

Commit f602867

Browse files
committed
fix default config for deepcompile
Signed-off-by: Masahiro Tanaka <[email protected]>
1 parent 781e116 commit f602867

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

deepspeed/compile/config.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
class CompileConfig(DeepSpeedConfigModel):
1010
""" Configure compile settings """
1111

12-
deepcompile: bool = True
12+
deepcompile: bool = False
1313
""" Turn on/off the DeepCompile mode """
1414

1515
free_activation: bool = False

deepspeed/runtime/engine.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -2041,7 +2041,7 @@ def forward(self, *inputs, **kwargs):
20412041
if self.autotuning_profile_model_info():
20422042
ma = get_ma_status()
20432043

2044-
if self.is_deepcompile_enabled():
2044+
if self.is_deepcompile_enabled() and hasattr(self, "launch_compile_passes"):
20452045
# We can't have this in forward prologue as the compiler compiles hooks including the forward prologue.
20462046
self.launch_compile_passes(self.global_steps)
20472047

0 commit comments

Comments
 (0)