[Mlir-commits] [mlir] [mlir][gpu] Allow subgroup reductions over 1-d vector types (PR #76015)
Fabian Mora
llvmlistbot at llvm.org
Wed Dec 20 07:44:30 PST 2023
https://github.com/fabianmcg commented:
Reducing GPU shuffles with a multi-reduce op only makes sense if the type has size less than 32 bits as both NVIDIA & AMD only support 32 bit shuffles.
So my question is, how do you plan to lower large vectors (ie. `vector<2xf32>`)?
I'd prefer only accepting `vector<1xi32>`, and let the user use `vector.bitcast`.
https://github.com/llvm/llvm-project/pull/76015
More information about the Mlir-commits
mailing list