@@ -78,18 +78,17 @@ TEST_P(urKernelGetSuggestedLocalWorkSizeTest, InvalidNullHandleQueue) {
78
78
79
79
TEST_P (urKernelGetSuggestedLocalWorkSizeTest, InvalidWorkDimension) {
80
80
uint32_t max_work_item_dimensions = 0 ;
81
- auto result = urDeviceGetInfo (
81
+ ASSERT_SUCCESS ( urDeviceGetInfo (
82
82
device, UR_DEVICE_INFO_MAX_WORK_ITEM_DIMENSIONS,
83
- sizeof (max_work_item_dimensions), &max_work_item_dimensions, nullptr );
83
+ sizeof (max_work_item_dimensions), &max_work_item_dimensions, nullptr ));
84
+ auto result = urKernelGetSuggestedLocalWorkSize (
85
+ kernel, queue, max_work_item_dimensions + 1 , &global_offset,
86
+ &global_size, &suggested_local_work_size);
87
+ ASSERT_SUCCESS (result);
84
88
if (result == UR_RESULT_ERROR_UNSUPPORTED_FEATURE) {
85
89
GTEST_SKIP ();
86
90
}
87
- ASSERT_SUCCESS (result);
88
- ASSERT_EQ_RESULT (urKernelGetSuggestedLocalWorkSize (
89
- kernel, queue, max_work_item_dimensions + 1 ,
90
- &global_offset, &global_size,
91
- &suggested_local_work_size),
92
- UR_RESULT_ERROR_INVALID_WORK_DIMENSION);
91
+ ASSERT_EQ_RESULT (result, UR_RESULT_ERROR_INVALID_WORK_DIMENSION);
93
92
}
94
93
95
94
TEST_P (urKernelGetSuggestedLocalWorkSizeTest, InvalidGlobalOffset) {
0 commit comments