summaryrefslogtreecommitdiff
path: root/mlir/python
diff options
context:
space:
mode:
authorLorenzo Chelini <l.chelini@icloud.com>2022-09-19 12:17:30 +0200
committerLorenzo Chelini <l.chelini@icloud.com>2022-09-19 12:17:30 +0200
commite9dd2b2d4b9ccf8ebbbb72bc8e0249c9296be5e4 (patch)
tree0d465957b5f2467e9129de32a36f2b6b772ca095 /mlir/python
parentf381768a8da6bd6bde8bdff34f080bf12bf20064 (diff)
downloadllvm-e9dd2b2d4b9ccf8ebbbb72bc8e0249c9296be5e4.tar.gz
Revert "[MLIR][Linalg] introduce batch-reduce GEMM"
This reverts commit f381768a8da6bd6bde8bdff34f080bf12bf20064.
Diffstat (limited to 'mlir/python')
-rw-r--r--mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py14
1 files changed, 0 insertions, 14 deletions
diff --git a/mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py b/mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py
index b9b292d847cd..983842cde132 100644
--- a/mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py
+++ b/mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py
@@ -150,20 +150,6 @@ def quantized_batch_matmul(A=TensorDef(T1, Batch, S.M, S.K),
TypeFn.cast_signed(U, AZp)) * (TypeFn.cast_signed(
U, B[D.b, D.k, D.n]) - TypeFn.cast_signed(U, BZp))
-@linalg_structured_op
-def batch_reduce_matmul(A=TensorDef(T1, Batch, S.M, S.K),
- B=TensorDef(T2, Batch, S.K, S.N),
- C=TensorDef(U, S.M, S.N, output=True)):
- """Performs a batch-reduce matrix multiplication of two 3D inputs.
- The partial multiplication results are reduced into a 2D output.
-
- Numeric casting is performed on the operands to the inner multiply, promoting
- them to the same data type as the accumulator/output.
- """
- domain(D.b, D.m, D.n, D.k)
- implements(ContractionOpInterface)
- C[D.m, D.n] += TypeFn.cast_signed(U, A[D.b, D.m, D.k] * TypeFn.cast_signed(
- U, B[D.b, D.k, D.n])
@linalg_structured_op
def matvec(A=TensorDef(T1, S.M, S.N),