@@ -180,6 +180,13 @@ struct urMemBufferTest : urContextTest {
180
180
struct urMemImageTest : urContextTest {
181
181
void SetUp () override {
182
182
UUR_RETURN_ON_FATAL_FAILURE (urContextTest::SetUp ());
183
+ ur_bool_t imageSupported = false ;
184
+ ASSERT_SUCCESS (
185
+ urDeviceGetInfo (this ->device , UR_DEVICE_INFO_IMAGE_SUPPORTED,
186
+ sizeof (ur_bool_t ), &imageSupported, nullptr ));
187
+ if (!imageSupported) {
188
+ GTEST_SKIP ();
189
+ }
183
190
}
184
191
185
192
void TearDown () override {
@@ -276,6 +283,13 @@ template <class T> struct urMemBufferTestWithParam : urContextTestWithParam<T> {
276
283
template <class T > struct urMemImageTestWithParam : urContextTestWithParam<T> {
277
284
void SetUp () override {
278
285
UUR_RETURN_ON_FATAL_FAILURE (urContextTestWithParam<T>::SetUp ());
286
+ ur_bool_t imageSupported = false ;
287
+ ASSERT_SUCCESS (
288
+ urDeviceGetInfo (this ->device , UR_DEVICE_INFO_IMAGE_SUPPORTED,
289
+ sizeof (ur_bool_t ), &imageSupported, nullptr ));
290
+ if (!imageSupported) {
291
+ GTEST_SKIP ();
292
+ }
279
293
ASSERT_SUCCESS (urMemImageCreate (this ->context , UR_MEM_FLAG_READ_WRITE,
280
294
&format, &desc, nullptr , &image));
281
295
ASSERT_NE (nullptr , image);
@@ -551,6 +565,13 @@ struct urMemBufferQueueTest : urQueueTest {
551
565
struct urMemImageQueueTest : urQueueTest {
552
566
void SetUp () override {
553
567
UUR_RETURN_ON_FATAL_FAILURE (urQueueTest::SetUp ());
568
+ ur_bool_t imageSupported = false ;
569
+ ASSERT_SUCCESS (
570
+ urDeviceGetInfo (this ->device , UR_DEVICE_INFO_IMAGE_SUPPORTED,
571
+ sizeof (ur_bool_t ), &imageSupported, nullptr ));
572
+ if (!imageSupported) {
573
+ GTEST_SKIP ();
574
+ }
554
575
ASSERT_SUCCESS (urMemImageCreate (this ->context , UR_MEM_FLAG_READ_WRITE,
555
576
&format, &desc1D, nullptr , &image1D));
556
577
@@ -626,6 +647,15 @@ struct urMemImageQueueTest : urQueueTest {
626
647
struct urMultiDeviceMemImageTest : urMultiDeviceContextTest {
627
648
void SetUp () override {
628
649
UUR_RETURN_ON_FATAL_FAILURE (urMultiDeviceContextTest::SetUp ());
650
+ for (auto device : DevicesEnvironment::instance->devices ) {
651
+ ur_bool_t imageSupported = false ;
652
+ ASSERT_SUCCESS (
653
+ urDeviceGetInfo (device, UR_DEVICE_INFO_IMAGE_SUPPORTED,
654
+ sizeof (ur_bool_t ), &imageSupported, nullptr ));
655
+ if (!imageSupported) {
656
+ GTEST_SKIP ();
657
+ }
658
+ }
629
659
ASSERT_SUCCESS (urMemImageCreate (context, UR_MEM_FLAG_READ_WRITE,
630
660
&format, &desc1D, nullptr , &image1D));
631
661
0 commit comments