debugging gpu test failures on ubi

GEOS-DEV · May 23, 2024 · ec9eef1 · ec9eef1
1 parent 602ba50
commit ec9eef1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/common/pmpl.hpp b/src/common/pmpl.hpp
@@ -137,8 +137,9 @@ void genericKernelWrapper( int const N, DATA_TYPE * & hostData, LAMBDA && func )
   genericKernel << < 1, 1 >> > ( std::forward< LAMBDA >( func ), deviceData );
   deviceDeviceSynchronize();
   deviceMemCpy( hostData, deviceData, N * sizeof(DATA_TYPE), cudaMemcpyDeviceToHost );
+  deviceFree( deviceData );
 #else
-  hostData = new DATA_TYPE[N];
+  SHIVA_UNUSED_VAR( N );
   genericKernel( std::forward< LAMBDA >( func ), hostData );
 #endif
 }