| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
[MLIR][Linalg] Remove matmul_transpose variants (#147961) Removes the (batch_)matmul_transpose_{a|b} variants from OpDSL and replace it with matmul affine_maps [...] whenever appropriate. This is in line with the [plan](https://discourse.llvm.org/t/rfc-op-explosion-in-linalg/82863), and can be done since #104783 merged. See: https://discourse.llvm.org/t/deprecate-batch-matmul-transpose-a-b-linalg-operations/87245 Issues investigated: * pad transform tests that could use matmul instead, so change to that. * ArmSME test using transpose actually needed it, so changed to matmul + affine maps. Arm tests validated by @banach-space (thanks!!). | 11 个月前 | |
[MLIR][Python] Add region_op wrappers for linalg (#167616) Makes linalg.reduce and linalg.map region_ops so they can be constructed from functions and be called as decorators. | 8 个月前 | |
[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> | 8 个月前 |