[SYCLomatic] Migrate CU_LAUNCH_PARAM_* macros (#652)

Signed-off-by: Lu, John <[email protected]>

Author: LU-JOHN

clang/lib/DPCT/MapNames.cpp

@@ -4163,6 +4163,19 @@ std::unordered_map<std::string, MacroMigrationRule> MapNames::MacroRuleMap{
     {"cudaStreamDefault",
      MacroMigrationRule("cudaStreamDefault_rule", RulePriority::Fallback,
                         "cudaStreamDefault", "0")},
+
+    {"CU_LAUNCH_PARAM_BUFFER_SIZE",
+     MacroMigrationRule("kernel_param_rule", RulePriority::Fallback,
+                        "CU_LAUNCH_PARAM_BUFFER_SIZE", "((void *) 2)",
+                        HelperFeatureEnum::Kernel_kernel_library)},
+    {"CU_LAUNCH_PARAM_BUFFER_POINTER",
+     MacroMigrationRule("kernel_param_rule", RulePriority::Fallback,
+                        "CU_LAUNCH_PARAM_BUFFER_POINTER", "((void *) 1)",
+                        HelperFeatureEnum::Kernel_kernel_library)},
+    {"CU_LAUNCH_PARAM_END",
+     MacroMigrationRule("kernel_param_rule", RulePriority::Fallback,
+                        "CU_LAUNCH_PARAM_END", "((void *) 0)",
+                        HelperFeatureEnum::Kernel_kernel_library)},
     //...
 };
 

clang/test/dpct/module_main.cu

@@ -1,6 +1,8 @@
 // RUN: dpct --format-range=none -out-root %T/module_main %s --cuda-include-path="%cuda-path/include" -- -x cuda --cuda-host-only
 // RUN: FileCheck %s --match-full-lines --input-file %T/module_main/module_main.dp.cpp
 
+//CHECK: #include <dpct/dpct.hpp>
+
 #include <string>
 int main(){
     //CHECK: dpct::kernel_library M;
@@ -28,6 +30,16 @@ int main(){
     //CHECK: F = dpct::get_kernel_function(M, FunctionName.c_str());
     cuModuleGetFunction(&F, M, FunctionName.c_str());
 
+
+    int    *argBuffer[3];
+    size_t  argBufferSize = sizeof(argBuffer);
+    //CHECK: void *extra[] = {((void *) 2), &argBufferSize,
+    //CHECK-NEXT: ((void *) 1), argBuffer,
+    //CHECK-NEXT: ((void *) 0)};
+    void   *extra[] = {CU_LAUNCH_PARAM_BUFFER_SIZE, &argBufferSize,
+                       CU_LAUNCH_PARAM_BUFFER_POINTER, argBuffer,
+                       CU_LAUNCH_PARAM_END};
+
     int sharedSize;
     CUstream s;
     void **param, **extra;