[Mlir-commits] [mlir] [MLIR][XeGPU] Lower vector.multi_reduction before linearization in XeGPUVectorLinearize (PR #190272)
Jianhui Li
llvmlistbot at llvm.org
Tue Apr 7 09:58:32 PDT 2026
================
@@ -265,4 +265,21 @@ gpu.module @test_kernel {
}
}
+// -----
+// CHECK-LABEL: func.func @test_vector_multi_reduction_add
+// CHECK-SAME: (%[[ARG0:.*]]: vector<16x1xf16>, %[[ARG1:.*]]: vector<1xf16>) -> vector<1xf16>
----------------
Jianhui-Li wrote:
The code look very inefficient to me. Why it is not just removing the training dimension and use a vector.reduction of 16 elements?
https://github.com/llvm/llvm-project/pull/190272
More information about the Mlir-commits
mailing list