SM60 supports scope on atom.cas

Author: akshayrdeodhar

llvm/lib/Target/NVPTX/NVPTXISelDAGToDAG.cpp

@@ -543,7 +543,7 @@ unsigned int NVPTXDAGToDAGISel::getMemOrder(const MemSDNode *N) const {
 
 unsigned int NVPTXDAGToDAGISel::getAtomicScope(const MemSDNode *N) const {
   // No "scope" modifier for SM/PTX versions which do not support scoped atomics
-  if (!Subtarget->hasAtomScope() || !Subtarget->hasMemoryOrdering())
+  if (!Subtarget->hasAtomScope())
     return NVPTX::Scope::Thread;
   return Scopes[N->getSyncScopeID()];
 }

llvm/test/CodeGen/NVPTX/atomicrmw-expand.ll

@@ -93,21 +93,26 @@ entry:
   %3 = atomicrmw or ptr %0, i8 %1 monotonic, align 1
   ; ALL: atom.xor.b32
   %4 = atomicrmw xor ptr %0, i8 %1 monotonic, align 1
-  ; ALL: atom.cas.b32
+  ; SM30: atom.cas.b32
+  ; SM60: atom.sys.cas.b32
   %5 = atomicrmw xchg ptr %0, i8 %1 monotonic, align 1
   ret void
 }
 
 ; CHECK-LABEL: minmax_i8
 define void @minmax_i8(ptr %0, i8 %1) {
 entry:
-  ; ALL: atom.cas.b32
+  ; SM30: atom.cas.b32
+  ; SM60: atom.sys.cas.b32
   %2 = atomicrmw min ptr %0, i8 %1 monotonic, align 1
-  ; ALL: atom.cas.b32
+  ; SM30: atom.cas.b32
+  ; SM60: atom.sys.cas.b32
   %3 = atomicrmw max ptr %0, i8 %1 monotonic, align 1
-  ; ALL: atom.cas.b32
+  ; SM30: atom.cas.b32
+  ; SM60: atom.sys.cas.b32
   %4 = atomicrmw umin ptr %0, i8 %1 monotonic, align 1
-  ; ALL: atom.cas.b32
+  ; SM30: atom.cas.b32
+  ; SM60: atom.sys.cas.b32
   %5 = atomicrmw umax ptr %0, i8 %1 monotonic, align 1
   ret void
 }
@@ -121,21 +126,26 @@ entry:
   %3 = atomicrmw or ptr %0, i16 %1 monotonic, align 2
   ; ALL: atom.xor.b32
   %4 = atomicrmw xor ptr %0, i16 %1 monotonic, align 2
-  ; ALL: atom.cas.b32
+  ; SM30: atom.cas.b32
+  ; SM60: atom.sys.cas.b32
   %5 = atomicrmw xchg ptr %0, i16 %1 monotonic, align 2
   ret void
 }
 
 ; CHECK-LABEL: minmax_i16
 define void @minmax_i16(ptr %0, i16 %1) {
 entry:
-  ; ALL: atom.cas.b32
+  ; SM30: atom.cas.b32
+  ; SM60: atom.sys.cas.b32
   %2 = atomicrmw min ptr %0, i16 %1 monotonic, align 2
-  ; ALL: atom.cas.b32
+  ; SM30: atom.cas.b32
+  ; SM60: atom.sys.cas.b32
   %3 = atomicrmw max ptr %0, i16 %1 monotonic, align 2
-  ; ALL: atom.cas.b32
+  ; SM30: atom.cas.b32
+  ; SM60: atom.sys.cas.b32
   %4 = atomicrmw umin ptr %0, i16 %1 monotonic, align 2
-  ; ALL: atom.cas.b32
+  ; SM30: atom.cas.b32
+  ; SM60: atom.sys.cas.b32
   %5 = atomicrmw umax ptr %0, i16 %1 monotonic, align 2
   ret void
 }