Skip to content

Commit 39fd0c8

Browse files
durga4githubgokulbmishra
authored andcommitted
[MLIR][NVVM] Add support for shared::cta destination (llvm#168056)
This patch adds support for shared::cta as destination space in the TMA non-tensor copy Op (from global to shared::cta). * Appropriate verifier checks are added. * Unit tests are added to verify the lowering. The related intrinsic changes were merged through PR llvm#167508. Signed-off-by: Durgadoss R <[email protected]>
1 parent fb8eacd commit 39fd0c8

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

mlir/lib/Dialect/LLVMIR/IR/NVVMDialect.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -249,6 +249,7 @@ LogicalResult CpAsyncBulkGlobalToSharedClusterOp::verify() {
249249
return success();
250250
}
251251

252+
<<<<<<< HEAD
252253
static LogicalResult verifyMBarrierArriveLikeOp(Operation *op, Value addr,
253254
NVVM::MemScopeKind scope,
254255
Value retVal = nullptr) {
@@ -282,6 +283,8 @@ LogicalResult MBarrierCompleteTxOp::verify() {
282283
return verifyMBarrierArriveLikeOp(getOperation(), getAddr(), getScope());
283284
}
284285

286+
=======
287+
>>>>>>> 95aa70cf209a ([MLIR][NVVM] Add support for shared::cta destination (#168056))
285288
LogicalResult ConvertFloatToTF32Op::verify() {
286289
using RndMode = NVVM::FPRoundingMode;
287290
switch (getRnd()) {

0 commit comments

Comments
 (0)