llvm-project
a5a78d0b - [mlir][linalg][python] Add Python Bindings for Inferring Contraction Dimensions from Affine Maps (#167587)

Commit
92 days ago
[mlir][linalg][python] Add Python Bindings for Inferring Contraction Dimensions from Affine Maps (#167587) This PR exposes `linalg::inferContractionDims(ArrayRef<AffineMap>)` to Python, allowing users to infer contraction dimensions (batch/m/n/k) directly from a list of affine maps without needing an operation. --------- Signed-off-by: Bangtian Liu <liubangtian@gmail.com>
Author
Parents
Loading