commit f6deb74e4b5305a50068332b0e9b2b40cfd9731d
author Jon Chesterfield <jonathanchesterfield@gmail.com> Sun Jan 24 18:30:04 2021 +0000
committer Copybara-Service <copybara-worker@google.com> Sun Jan 24 21:48:08 2021 -0800
tree b79e84c7ef79eb313f999ac6d434bf0017ca35f6
parent cbb2d4ea57c0b812dcf706c233e4677a2666ca09

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

GitOrigin-RevId: e5e448aafa7699c17f78aaffb001b665b607e5ae

libomptarget/plugins/cuda/dynamic_cuda/cuda.cpp

diff --git a/libomptarget/plugins/cuda/dynamic_cuda/cuda.cpp b/libomptarget/plugins/cuda/dynamic_cuda/cuda.cpp
index cc7bc42..ad67fe9 100644
--- a/libomptarget/plugins/cuda/dynamic_cuda/cuda.cpp
+++ b/libomptarget/plugins/cuda/dynamic_cuda/cuda.cpp
@@ -28,26 +28,26 @@
 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);