netbird/gvisor
0
0
Fork 0
mirror of https://github.com/netbirdio/gvisor.git synced 2026-05-22 17:12:49 -07:00
Code Issues Packages Projects Releases Wiki Activity

Check for CUDA kernel launch errors in GPU smoke test.

Browse Source 
PiperOrigin-RevId: 694699470
This commit is contained in:
Jamie Liu
2024-11-08 17:34:13 -08:00
committed by gVisor bot
parent f37f2c37c8
commit 7c2bcddc13
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
images/gpu/cuda-tests/cuda_malloc.cu
+5
View File
@@ -68,6 +68,7 @@ void TestMallocHostReadWrite(int device) {
// Copy the array to the device, mutate it there, and copy it back.
CHECK_CUDA(cudaMemcpy(gpu_data, cpu_data, kNumBytes, cudaMemcpyHostToDevice));
addKernel<<<kNumBlocks, kNumThreads>>>(gpu_data);
CHECK_CUDA(cudaGetLastError());
CHECK_CUDA(cudaDeviceSynchronize());
CHECK_CUDA(cudaMemcpy(cpu_data, gpu_data, kNumBytes, cudaMemcpyDeviceToHost));
@@ -111,6 +112,7 @@ void TestMallocHostReadWrite(int device) {
// Mutate the array on the device again.
CHECK_CUDA(cudaMemcpy(gpu_data, cpu_data, kNumBytes, cudaMemcpyHostToDevice));
addKernel<<<kNumBlocks, kNumThreads>>>(gpu_data);
CHECK_CUDA(cudaGetLastError());
CHECK_CUDA(cudaDeviceSynchronize());
CHECK_CUDA(cudaMemcpy(cpu_data, gpu_data, kNumBytes, cudaMemcpyDeviceToHost));
@@ -153,6 +155,7 @@ void TestMallocManagedRoundTrip(int device, unsigned int malloc_flags,
// Mutate the array on the device.
addKernel<<<kNumBlocks, kNumThreads>>>(data);
CHECK_CUDA(cudaGetLastError());
CHECK_CUDA(cudaDeviceSynchronize());
if (prefetch) {
@@ -208,6 +211,7 @@ void TestMallocManagedReadWrite(int device) {
// Mutate the array on the device.
addKernel<<<kNumBlocks, kNumThreads>>>(data);
CHECK_CUDA(cudaGetLastError());
CHECK_CUDA(cudaDeviceSynchronize());
// Check that the array has the expected result.
@@ -249,6 +253,7 @@ void TestMallocManagedReadWrite(int device) {
// Mutate the array on the device again.
addKernel<<<kNumBlocks, kNumThreads>>>(data);
CHECK_CUDA(cudaGetLastError());
CHECK_CUDA(cudaDeviceSynchronize());
// Check that the array has the expected result again.