Fix optimize string (#1160)

Author: wsmoses

deps/build_local.jl

@@ -137,6 +137,10 @@ push!(build_cmd_list, "--repo_env=CC=$(cc)")
 push!(build_cmd_list, "--check_visibility=false")
 push!(build_cmd_list, "--verbose_failures")
 push!(build_cmd_list, "--jobs=$(parsed_args["jobs"])")
+if Sys.isapple()
+    push!(build_cmd_list, "--define")
+    push!(build_cmd_list, "using_clang=true")
+end
 for opt in parsed_args["copt"]
     push!(build_cmd_list, "--copt=$(opt)")
 end

src/Compiler.jl

@@ -1292,23 +1292,25 @@ function compile_mlir!(
     end
 
     # HACK: remove with next JLL
-    if transpose_propagate === :up
-        run_pass_pipeline!(
-            mod,
-            "enzyme-hlo-generate-td{patterns=transpose_while},transform-interpreter,enzyme-hlo-remove-transform",
-            "transpose_while",
-        )
-    end
+    if !(optimize isa String)
+        if transpose_propagate === :up
+            run_pass_pipeline!(
+                mod,
+                "enzyme-hlo-generate-td{patterns=transpose_while},transform-interpreter,enzyme-hlo-remove-transform",
+                "transpose_while",
+            )
+        end
 
-    if optimize ∉ (:none, :just_batch, :canonicalize) &&
-        (transpose_propagate === :up || reshape_propagate === :up)
-        # We tried propagating reshapes and transposes up. If at this point we are left with
-        # them, we propagate them down to minimize the number of Ops in the IR.
-        run_pass_pipeline!(
-            mod,
-            optimization_passes(; transpose_propagate=:down, reshape_propagate=:down),
-            "post_op_transpose_reshape",
-        )
+        if optimize ∉ (:none, :just_batch, :canonicalize) &&
+            (transpose_propagate === :up || reshape_propagate === :up)
+            # We tried propagating reshapes and transposes up. If at this point we are left with
+            # them, we propagate them down to minimize the number of Ops in the IR.
+            run_pass_pipeline!(
+                mod,
+                optimization_passes(; transpose_propagate=:down, reshape_propagate=:down),
+                "post_op_transpose_reshape",
+            )
+        end
     end
 
     # Now we resolve paddings if `optimize_then_pad`