diff --git a/catch/unit/memory/hipArrayGetDescriptor.cc b/catch/unit/memory/hipArrayGetDescriptor.cc index 608aef1c90..c977590987 100644 --- a/catch/unit/memory/hipArrayGetDescriptor.cc +++ b/catch/unit/memory/hipArrayGetDescriptor.cc @@ -236,6 +236,7 @@ TEST_CASE("Unit_hipArrayGetDescriptor_1D_2D_ArrayParameterChk") { HIP_CHECK(hipGetDeviceCount(&numDevices)); for (int i = 0; i < numDevices; i++) { HIP_CHECK(hipSetDevice(i)); + CHECK_IMAGE_SUPPORT #if HT_NVIDIA HIP_CHECK(hipInit(0)); hipCtx_t ctx; @@ -328,6 +329,7 @@ TEST_CASE("Unit_hipArrayGetDescriptor_MultiThreadScenarioFor1D_2D_Array") { HIP_CHECK(hipGetDeviceCount(&numDevices)); for (int i = 0; i < numDevices; i++) { HIP_CHECK(hipSetDevice(i)); + CHECK_IMAGE_SUPPORT #if HT_NVIDIA HIP_CHECK(hipInit(0)); hipCtx_t ctx; @@ -376,6 +378,7 @@ TEST_CASE("Unit_hipArrayGetDescriptor_Host2Array_Array2Host") { HIP_CHECK(hipGetDeviceCount(&numDevices)); for (int k = 0; k < numDevices; k++) { HIP_CHECK(hipSetDevice(k)); + CHECK_IMAGE_SUPPORT #if HT_NVIDIA HIP_CHECK(hipInit(0)); hipCtx_t ctx;