Skip to content

Commit d63e4c0

Browse files
wsmosesgiordanogithub-actions[bot]
authored
Fix optimize_comms=false (#1437)
* Fix optimize_comms=false * Update src/Compiler.jl Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> * Update Compiler.jl --------- Co-authored-by: Mosè Giordano <[email protected]> Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
1 parent 9c6cbf7 commit d63e4c0

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/Compiler.jl

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1447,7 +1447,13 @@ function raising!(f, is_raising::Bool)
14471447
end
14481448

14491449
function get_optimize_comms_passes(options::Bool)
1450-
options || return String[]
1450+
if !options
1451+
return [
1452+
"enzyme-hlo-generate-td{patterns=lower_rotate;lower_wrap;lower_extend}",
1453+
"transform-interpreter",
1454+
"enzyme-hlo-remove-transform",
1455+
]
1456+
end
14511457
return get_optimize_comms_passes(OptimizeCommunicationOptions())
14521458
end
14531459

0 commit comments

Comments
 (0)