update

Author: minansys

enzyme/Enzyme/AdjointGenerator.h

@@ -3996,12 +3996,6 @@ class AdjointGenerator : public llvm::InstVisitor<AdjointGenerator> {
       (void)vdiff;
 
       switch (ID) {
-#if LLVM_VERSION_MAJOR <= 20
-      case Intrinsic::nvvm_barrier0:
-#else
-      case Intrinsic::nvvm_barrier_cta_sync_aligned_all:
-      case Intrinsic::nvvm_barrier_cta_sync_aligned_count:
-#endif
 #if LLVM_VERSION_MAJOR < 22
       case Intrinsic::nvvm_barrier0_popc:
       case Intrinsic::nvvm_barrier0_and:
@@ -4013,6 +4007,32 @@ class AdjointGenerator : public llvm::InstVisitor<AdjointGenerator> {
       case Intrinsic::nvvm_barrier_cta_red_or_aligned_count:
       case Intrinsic::nvvm_barrier_cta_red_popc_aligned_all:
       case Intrinsic::nvvm_barrier_cta_red_popc_aligned_count:
+#endif
+      {
+        SmallVector<Value *, 1> args = {};
+#if LLVM_VERSION_MAJOR > 20
+        auto cal = cast<CallInst>(Builder2.CreateCall(
+            getIntrinsicDeclaration(
+                M, Intrinsic::nvvm_barrier_cta_sync_aligned_all),
+            args));
+        cal->setCallingConv(getIntrinsicDeclaration(
+                                M, Intrinsic::nvvm_barrier_cta_sync_aligned_all)
+                                ->getCallingConv());
+#else
+        auto cal = cast<CallInst>(Builder2.CreateCall(
+            getIntrinsicDeclaration(M, Intrinsic::nvvm_barrier0), args));
+        cal->setCallingConv(getIntrinsicDeclaration(M, Intrinsic::nvvm_barrier0)
+                                ->getCallingConv());
+#endif
+        cal->setDebugLoc(gutils->getNewFromOriginal(I.getDebugLoc()));
+        return false;
+      }
+
+#if LLVM_VERSION_MAJOR <= 20
+      case Intrinsic::nvvm_barrier0:
+#else
+      case Intrinsic::nvvm_barrier_cta_sync_aligned_all:
+      case Intrinsic::nvvm_barrier_cta_sync_aligned_count:
 #endif
       case Intrinsic::amdgcn_s_barrier:
       case Intrinsic::nvvm_membar_cta: