@@ -662,4 +662,152 @@ def XeGPU_UpdateOffsetOp: XeGPU_Op<"update_offset",
}];
}
+def XeGPU_DpasOp : XeGPU_Op<"dpas", [Pure, AllElementTypesMatch<["lhs",
"rhs"]>]> {
+ let summary = "It performs mma computation";
+
+ let description = [{DPAS performs matrix multiplication
@@ -662,4 +662,152 @@ def XeGPU_UpdateOffsetOp: XeGPU_Op<"update_offset",
}];
}
+def XeGPU_DpasOp : XeGPU_Op<"dpas", [Pure, AllElementTypesMatch<["lhs",
"rhs"]>]> {
+ let summary = "It performs mma computation";
+
+ let description = [{DPAS performs matrix multiplication
@@ -662,4 +662,152 @@ def XeGPU_UpdateOffsetOp: XeGPU_Op<"update_offset",
}];
}
+def XeGPU_DpasOp : XeGPU_Op<"dpas", [Pure, AllElementTypesMatch<["lhs",
"rhs"]>]> {
+ let summary = "It performs mma computation";
+
+ let description = [{DPAS performs matrix multiplication
llvmbot wrote:
@llvm/pr-subscribers-mlir
Author: Chao Chen (chencha3)
Changes
This PR adds definition of dpas op, atomic op, named barrier and related ops.
---
Full diff: https://github.com/llvm/llvm-project/pull/88439.diff
8 Files Affected:
- (modified)
llvmbot wrote:
@llvm/pr-subscribers-mlir-gpu
Author: Chao Chen (chencha3)
Changes
This PR adds definition of dpas op, atomic op, named barrier and related ops.
---
Full diff: https://github.com/llvm/llvm-project/pull/88439.diff
8 Files Affected:
- (modified)
https://github.com/chencha3 created
https://github.com/llvm/llvm-project/pull/88439
This PR adds definition of dpas op, atomic op, named barrier and related ops.
>From 6021411059863c9a2bfdfc91e35628328e709a8c Mon Sep 17 00:00:00 2001
From: Chao Chen
Date: Thu, 11 Apr 2024 15:46:26 -0500