diff --git a/sycl/plugins/cuda/pi_cuda.cpp b/sycl/plugins/cuda/pi_cuda.cpp index fc801df260c93..a5749f4ca0cbd 100644 --- a/sycl/plugins/cuda/pi_cuda.cpp +++ b/sycl/plugins/cuda/pi_cuda.cpp @@ -4164,6 +4164,9 @@ pi_result cuda_piextUSMHostAlloc(void **result_ptr, pi_context context, result = error; } + assert(alignment == 0 || + (result == PI_SUCCESS && + reinterpret_cast(*result_ptr) % alignment == 0)); return result; } @@ -4185,6 +4188,9 @@ pi_result cuda_piextUSMDeviceAlloc(void **result_ptr, pi_context context, result = error; } + assert(alignment == 0 || + (result == PI_SUCCESS && + reinterpret_cast(*result_ptr) % alignment == 0)); return result; } @@ -4207,6 +4213,9 @@ pi_result cuda_piextUSMSharedAlloc(void **result_ptr, pi_context context, result = error; } + assert(alignment == 0 || + (result == PI_SUCCESS && + reinterpret_cast(*result_ptr) % alignment == 0)); return result; } diff --git a/sycl/plugins/level_zero/pi_level_zero.cpp b/sycl/plugins/level_zero/pi_level_zero.cpp index efbf971d6945b..a970ef387faa8 100644 --- a/sycl/plugins/level_zero/pi_level_zero.cpp +++ b/sycl/plugins/level_zero/pi_level_zero.cpp @@ -4674,6 +4674,8 @@ pi_result piextUSMHostAlloc(void **ResultPtr, pi_context Context, ZE_CALL( zeMemAllocHost(Context->ZeContext, &ZeDesc, Size, Alignment, ResultPtr)); + assert(Alignment == 0 || + reinterpret_cast(*ResultPtr) % Alignment == 0); return PI_SUCCESS; } @@ -4700,6 +4702,8 @@ pi_result USMDeviceAllocImpl(void **ResultPtr, pi_context Context, ZE_CALL(zeMemAllocDevice(Context->ZeContext, &ZeDesc, Size, Alignment, Device->ZeDevice, ResultPtr)); + assert(Alignment == 0 || + reinterpret_cast(*ResultPtr) % Alignment == 0); return PI_SUCCESS; } @@ -4721,6 +4725,8 @@ pi_result USMSharedAllocImpl(void **ResultPtr, pi_context Context, ZE_CALL(zeMemAllocShared(Context->ZeContext, &ZeDevDesc, &ZeHostDesc, Size, Alignment, Device->ZeDevice, ResultPtr)); + assert(Alignment == 0 || + reinterpret_cast(*ResultPtr) % Alignment == 0); return PI_SUCCESS; } diff --git a/sycl/plugins/opencl/pi_opencl.cpp b/sycl/plugins/opencl/pi_opencl.cpp index 1aae5013a92a8..6a65cc6e9962d 100644 --- a/sycl/plugins/opencl/pi_opencl.cpp +++ b/sycl/plugins/opencl/pi_opencl.cpp @@ -713,6 +713,9 @@ pi_result piextUSMHostAlloc(void **result_ptr, pi_context context, *result_ptr = Ptr; + assert(alignment == 0 || + (RetVal == PI_SUCCESS && + reinterpret_cast(*result_ptr) % alignment == 0)); return RetVal; } @@ -746,6 +749,9 @@ pi_result piextUSMDeviceAlloc(void **result_ptr, pi_context context, *result_ptr = Ptr; + assert(alignment == 0 || + (RetVal == PI_SUCCESS && + reinterpret_cast(*result_ptr) % alignment == 0)); return RetVal; } @@ -779,6 +785,9 @@ pi_result piextUSMSharedAlloc(void **result_ptr, pi_context context, *result_ptr = Ptr; + assert(alignment == 0 || + (RetVal == PI_SUCCESS && + reinterpret_cast(*result_ptr) % alignment == 0)); return RetVal; }