[Mlir-commits] [mlir] [mlir][Linalg] Allow PartialReductionOpInterface ops in tile_reduction_using_for (PR #120118)
Kunwar Grover
llvmlistbot at llvm.org
Mon Dec 23 05:07:45 PST 2024
Groverkss wrote:
> We could/should do that maybe. But I am fine with not doing that as well. We could make something like `linalg.softmax` implement this interface.
linalg.softmax cannot implement this interface because the reduction dimension is not tilable... Maybe we could add a online_softmax op for testing more interesting stuff, but maybe later.
https://github.com/llvm/llvm-project/pull/120118
More information about the Mlir-commits
mailing list