diff options
author | Lorenzo Chelini <l.chelini@icloud.com> | 2022-09-19 12:17:30 +0200 |
---|---|---|
committer | Lorenzo Chelini <l.chelini@icloud.com> | 2022-09-19 12:17:30 +0200 |
commit | e9dd2b2d4b9ccf8ebbbb72bc8e0249c9296be5e4 (patch) | |
tree | 0d465957b5f2467e9129de32a36f2b6b772ca095 /mlir/python | |
parent | f381768a8da6bd6bde8bdff34f080bf12bf20064 (diff) | |
download | llvm-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.py | 14 |
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), |