[llvm-branch-commits] [openmp] e5e448a - [libomptarget][cuda] Fix build, change missed from D95274

[Jon Chesterfield via llvm-branch-commits]

Author: Jon Chesterfield
Date: 2021-01-24T18:30:04Z
New Revision: e5e448aafa7699c17f78aaffb001b665b607e5ae

URL: https://github.com/llvm/llvm-project/commit/e5e448aafa7699c17f78aaffb001b665b607e5ae
DIFF: https://github.com/llvm/llvm-project/commit/e5e448aafa7699c17f78aaffb001b665b607e5ae.diff

LOG: [libomptarget][cuda] Fix build, change missed from D95274

Added: 
    

Modified: 
    openmp/libomptarget/plugins/cuda/dynamic_cuda/cuda.cpp

Removed: 
    


################################################################################
diff  --git a/openmp/libomptarget/plugins/cuda/dynamic_cuda/cuda.cpp b/openmp/libomptarget/plugins/cuda/dynamic_cuda/cuda.cpp
index cc7bc42412f6..ad67fe95c77e 100644
--- a/openmp/libomptarget/plugins/cuda/dynamic_cuda/cuda.cpp
+++ b/openmp/libomptarget/plugins/cuda/dynamic_cuda/cuda.cpp
@@ -28,26 +28,26 @@ DLWRAP(cuFuncGetAttribute, 3);
 DLWRAP(cuGetErrorString, 2);
 DLWRAP(cuLaunchKernel, 11);
 
-DLWRAP(cuMemAlloc, 2);
-DLWRAP(cuMemcpyDtoDAsync, 4);
+DLWRAP(cuMemAlloc_v2, 2);
+DLWRAP(cuMemcpyDtoDAsync_v2, 4);
 
-DLWRAP(cuMemcpyDtoH, 3);
-DLWRAP(cuMemcpyDtoHAsync, 4);
-DLWRAP(cuMemcpyHtoD, 3);
-DLWRAP(cuMemcpyHtoDAsync, 4);
+DLWRAP(cuMemcpyDtoH_v2, 3);
+DLWRAP(cuMemcpyDtoHAsync_v2, 4);
+DLWRAP(cuMemcpyHtoD_v2, 3);
+DLWRAP(cuMemcpyHtoDAsync_v2, 4);
 
-DLWRAP(cuMemFree, 1);
+DLWRAP(cuMemFree_v2, 1);
 DLWRAP(cuModuleGetFunction, 3);
-DLWRAP(cuModuleGetGlobal, 4);
+DLWRAP(cuModuleGetGlobal_v2, 4);
 
 DLWRAP(cuModuleUnload, 1);
 DLWRAP(cuStreamCreate, 2);
-DLWRAP(cuStreamDestroy, 1);
+DLWRAP(cuStreamDestroy_v2, 1);
 DLWRAP(cuStreamSynchronize, 1);
 DLWRAP(cuCtxSetCurrent, 1);
-DLWRAP(cuDevicePrimaryCtxRelease, 1);
+DLWRAP(cuDevicePrimaryCtxRelease_v2, 1);
 DLWRAP(cuDevicePrimaryCtxGetState, 3);
-DLWRAP(cuDevicePrimaryCtxSetFlags, 2);
+DLWRAP(cuDevicePrimaryCtxSetFlags_v2, 2);
 DLWRAP(cuDevicePrimaryCtxRetain, 2);
 DLWRAP(cuModuleLoadDataEx, 5);