Skip to content

Commit 6faf68e

Browse files
authored
[Compile] Reset custom allreduce flag in O2 mode to NONE (#3081)
This PR resets the default IPC allreduce strategy under O2 mode to NONE for platform compatability.
1 parent 3463ab7 commit 6faf68e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/mlc_llm/interface/compiler_flags.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -210,7 +210,7 @@ def from_str(source: str) -> "ModelConfigOverride":
210210
faster_transformer=False,
211211
cudagraph=True,
212212
cutlass=True,
213-
ipc_allreduce_strategy=IPCAllReduceStrategyType.AUTO,
213+
ipc_allreduce_strategy=IPCAllReduceStrategyType.NONE,
214214
),
215215
"O3": OptimizationFlags(
216216
flashinfer=True,

0 commit comments

Comments
 (0)