Added migration logic for RMA, RMA_NBI, Memory Ord, Signal & Collective ops

Author: TejaX-Alaghari

clang/lib/DPCT/RuleInfra/MapNames.cpp

@@ -907,11 +907,16 @@ void MapNames::setExplicitNamespaceMap(
       {"cudaExternalSemaphoreHandleType",
        std::make_shared<TypeNameRule>(getExpNamespace() +
                                       "external_semaphore_handle_type")},
+      // nvSHMEM
       {"nvshmem_team_t", std::make_shared<TypeNameRule>("ishmem_team_t")},
       {"nvshmem_team_config_t",
        std::make_shared<TypeNameRule>("ishmem_team_config_t")},
       {"nvshmemx_init_attr_t",
        std::make_shared<TypeNameRule>("ishmemx_attr_t")},
+      {"nvshmemi_amo_t",
+       std::make_shared<TypeNameRule>("int")},
+      {"nvshmemi_cmp_type",
+       std::make_shared<TypeNameRule>("int")},
       // ...
   };
   // SYCLcompat unsupport types
@@ -1601,6 +1606,7 @@ void MapNames::setExplicitNamespaceMap(
                ? getExpNamespace() +
                      "external_semaphore_handle_type::win32_nt_dx12_fence"
                : "cudaExternalSemaphoreHandleTypeD3D12Fence")},
+      // nvSHMEM
       {"NVSHMEM_TEAM_WORLD",
        std::make_shared<EnumNameRule>("ISHMEM_TEAM_WORLD")},
       {"NVSHMEM_TEAM_SHARED",
@@ -1611,6 +1617,22 @@ void MapNames::setExplicitNamespaceMap(
        std::make_shared<EnumNameRule>("ISHMEMX_RUNTIME_MPI")},
       {"NVSHMEMX_INIT_WITH_SHMEM",
        std::make_shared<EnumNameRule>("ISHMEMX_RUNTIME_OPENSHMEM")},
+      {"NVSHMEM_SIGNAL_SET",
+       std::make_shared<EnumNameRule>("ISHMEM_SIGNAL_SET")},
+      {"NVSHMEM_SIGNAL_ADD",
+       std::make_shared<EnumNameRule>("ISHMEM_SIGNAL_ADD")},
+      {"NVSHMEM_CMP_EQ",
+       std::make_shared<EnumNameRule>("ISHMEM_CMP_EQ")},
+      {"NVSHMEM_CMP_NE",
+       std::make_shared<EnumNameRule>("ISHMEM_CMP_NE")},
+      {"NVSHMEM_CMP_GT",
+       std::make_shared<EnumNameRule>("ISHMEM_CMP_GT")},
+      {"NVSHMEM_CMP_GE",
+       std::make_shared<EnumNameRule>("ISHMEM_CMP_GE")},
+      {"NVSHMEM_CMP_LT",
+       std::make_shared<EnumNameRule>("ISHMEM_CMP_LT")},
+      {"NVSHMEM_CMP_LE",
+       std::make_shared<EnumNameRule>("ISHMEM_CMP_LE")},
       // ...
   };
 

clang/lib/DPCT/RulesLang/RulesLang.cpp

@@ -348,7 +348,7 @@ void TypeInDeclRule::registerMatcher(MatchFinder &MF) {
               "cudaGraphicsRegisterFlags", "cudaExternalMemoryHandleType",
               "cudaExternalSemaphoreHandleType", "CUstreamCallback",
               "cudaHostFn_t", "__nv_half2", "__nv_half",
-              "cudaGraphNodeType"))))))
+              "cudaGraphNodeType", "nvshmemi_amo_t", "nvshmemi_cmp_type"))))))
           .bind("cudaTypeDef"),
       this);