Skip to content

Commit 205953d

Browse files
committed
Check error codes from getArrayDesc
1 parent 0942e74 commit 205953d

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

source/adapters/hip/enqueue.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1017,7 +1017,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urEnqueueMemImageRead(
10171017

10181018
hipArray_Format Format;
10191019
size_t NumChannels;
1020-
getArrayDesc(Array, Format, NumChannels);
1020+
UR_CHECK_ERROR(getArrayDesc(Array, Format, NumChannels));
10211021

10221022
int ElementByteSize = imageElementByteSize(Format);
10231023

@@ -1078,7 +1078,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urEnqueueMemImageWrite(
10781078

10791079
hipArray_Format Format;
10801080
size_t NumChannels;
1081-
getArrayDesc(Array, Format, NumChannels);
1081+
UR_CHECK_ERROR(getArrayDesc(Array, Format, NumChannels));
10821082

10831083
int ElementByteSize = imageElementByteSize(Format);
10841084

@@ -1141,13 +1141,13 @@ UR_APIEXPORT ur_result_t UR_APICALL urEnqueueMemImageCopy(
11411141
std::get<SurfaceMem>(hImageSrc->Mem).getArray(hQueue->getDevice());
11421142
hipArray_Format SrcFormat;
11431143
size_t SrcNumChannels;
1144-
getArrayDesc(SrcArray, SrcFormat, SrcNumChannels);
1144+
UR_CHECK_ERROR(getArrayDesc(SrcArray, SrcFormat, SrcNumChannels));
11451145

11461146
hipArray *DstArray =
11471147
std::get<SurfaceMem>(hImageDst->Mem).getArray(hQueue->getDevice());
11481148
hipArray_Format DstFormat;
11491149
size_t DstNumChannels;
1150-
getArrayDesc(DstArray, DstFormat, DstNumChannels);
1150+
UR_CHECK_ERROR(getArrayDesc(DstArray, DstFormat, DstNumChannels));
11511151

11521152
UR_ASSERT(SrcFormat == DstFormat,
11531153
UR_RESULT_ERROR_INVALID_IMAGE_FORMAT_DESCRIPTOR);

source/adapters/hip/kernel.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -282,7 +282,7 @@ urKernelSetArgMemObj(ur_kernel_handle_t hKernel, uint32_t argIndex,
282282
auto array = std::get<SurfaceMem>(hArgValue->Mem).getArray(Device);
283283
hipArray_Format Format;
284284
size_t NumChannels;
285-
getArrayDesc(array, Format, NumChannels);
285+
UR_CHECK_ERROR(getArrayDesc(array, Format, NumChannels));
286286
if (Format != HIP_AD_FORMAT_UNSIGNED_INT32 &&
287287
Format != HIP_AD_FORMAT_SIGNED_INT32 &&
288288
Format != HIP_AD_FORMAT_HALF && Format != HIP_AD_FORMAT_FLOAT) {

0 commit comments

Comments
 (0)