Update engine.py

microsoft · Sep 8, 2024 · 1913609 · 1913609
1 parent fc22d96
commit 1913609
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/runtime/engine.py b/deepspeed/runtime/engine.py
@@ -3675,7 +3675,7 @@ def compile(self, backend=get_accelerator().get_compile_backend(), compile_kwarg
         if self.is_compiled:
             return
 
-        self.module.compile(backend=backend, **compile_kwargs)
+        self.module.compile(**{compile_kwargs, 'backend': backend})
         self._is_compiled = True
 
     @property