use new codegen pass in LLVMGPU instead of linalgExtPass

Signed-off-by: Max Dawkins <[email protected]>
iree-org · Aug 9, 2024 · f4cd9c4 · f4cd9c4
1 parent f855a46
commit f4cd9c4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/compiler/src/iree/compiler/Codegen/LLVMGPU/Passes.cpp b/compiler/src/iree/compiler/Codegen/LLVMGPU/Passes.cpp
@@ -1054,8 +1054,8 @@ static void buildLLVMGPUCodegenConfigurationPassPipelineImpl(
     OpPassManager &modulePassManager) {
   {
     FunctionLikeNest funcPassManager(modulePassManager);
-    funcPassManager.addPredicatedPass(
-        clLLVMGPUUseIgemm, IREE::LinalgExt::createConvertConv2DToIm2ColOpPass);
+    funcPassManager.addPredicatedPass(clLLVMGPUUseIgemm,
+                                      createConvolutionToIGEMMPass());
     funcPassManager.addPass(createGPUGeneralizeNamedOpsPass);
     addCommonTargetExecutablePreprocessingPasses(funcPassManager);
     addEncodingToNopPasses(funcPassManager);