Metal: Run ExpandReductions pass. (#549)

Author: maleadt

Project.toml

@@ -19,7 +19,7 @@ InteractiveUtils = "1"
 Libdl = "1"
 Logging = "1"
 UUIDs = "1"
-LLVM = "6.4"
+LLVM = "6.6"
 Scratch = "1"
 TimerOutputs = "0.5"
 julia = "1.8"

src/metal.jl

@@ -182,6 +182,13 @@ function finish_ir!(@nospecialize(job::CompilerJob{MetalCompilerTarget}), mod::L
         end
     end
 
+    # perform codegen passes that would normally run during machine code emission
+    # XXX: codegen passes don't seem available in the new pass manager yet
+    @dispose pm=ModulePassManager() begin
+        expand_reductions!(pm)
+        run!(pm, mod)
+    end
+
     return functions(mod)[entry_fn]
 end