[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