Skip to content

Commit 3e94066

Browse files
authored
[UR][Offload] Make some more tests pass (#20216)
Some low hanging fruit for pass rate.
1 parent 579299b commit 3e94066

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

unified-runtime/source/adapters/offload/device.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,8 @@ UR_APIEXPORT ur_result_t UR_APICALL urDeviceGetInfo(ur_device_handle_t hDevice,
223223
case UR_DEVICE_INFO_ESIMD_SUPPORT:
224224
case UR_DEVICE_INFO_SUB_GROUP_INDEPENDENT_FORWARD_PROGRESS:
225225
case UR_DEVICE_INFO_ERROR_CORRECTION_SUPPORT:
226+
case UR_DEVICE_INFO_USM_P2P_SUPPORT_EXP:
227+
case UR_DEVICE_INFO_USM_CONTEXT_MEMCPY_SUPPORT_EXP:
226228
// TODO: Atomic queries in Offload
227229
case UR_DEVICE_INFO_ATOMIC_64:
228230
case UR_DEVICE_INFO_IMAGE_SRGB:

unified-runtime/source/adapters/offload/enqueue.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -157,6 +157,9 @@ UR_APIEXPORT ur_result_t UR_APICALL urEnqueueKernelLaunch(
157157
const size_t *pLocalWorkSize, uint32_t, const ur_kernel_launch_property_t *,
158158
uint32_t numEventsInWaitList, const ur_event_handle_t *phEventWaitList,
159159
ur_event_handle_t *phEvent) {
160+
UR_ASSERT(workDim > 0, UR_RESULT_ERROR_INVALID_WORK_DIMENSION);
161+
UR_ASSERT(workDim < 4, UR_RESULT_ERROR_INVALID_WORK_DIMENSION);
162+
160163
ol_queue_handle_t Queue;
161164
OL_RETURN_ON_ERR(hQueue->nextQueue(Queue));
162165
OL_RETURN_ON_ERR(waitOnEvents(Queue, phEventWaitList, numEventsInWaitList));

0 commit comments

Comments
 (0)