@@ -427,8 +427,10 @@ TEST_F(DeviceGetCapsTest, givenOpenCLVersion21WhenCapsAreCreatedThenDeviceReport
427427 }
428428 if (hwInfo->capabilityTable .supportsImages ) {
429429 EXPECT_THAT (caps.deviceExtensions , testing::HasSubstr (std::string (" cl_intel_spirv_media_block_io" )));
430+ EXPECT_THAT (caps.deviceExtensions , testing::HasSubstr (std::string (" cl_khr_3d_image_writes" )));
430431 } else {
431432 EXPECT_THAT (caps.deviceExtensions , testing::Not (testing::HasSubstr (std::string (" cl_intel_spirv_media_block_io" ))));
433+ EXPECT_THAT (caps.deviceExtensions , testing::Not (std::string (" cl_khr_3d_image_writes" )));
432434 }
433435 EXPECT_THAT (caps.deviceExtensions , testing::HasSubstr (std::string (" cl_intel_spirv_subgroups" )));
434436 EXPECT_THAT (caps.deviceExtensions , testing::HasSubstr (std::string (" cl_khr_spirv_no_integer_wrap_decoration" )));
@@ -454,6 +456,23 @@ TEST_F(DeviceGetCapsTest, givenNotSupportImagesWhenCapsAreCreatedThenDeviceNotRe
454456 EXPECT_THAT (caps.deviceExtensions , testing::Not (testing::HasSubstr (std::string (" cl_intel_spirv_media_block_io" ))));
455457}
456458
459+ TEST_F (DeviceGetCapsTest, givenSupportImagesWhenCapsAreCreatedThenDeviceReportsClKhr3dImageWritesExtensions) {
460+ HardwareInfo hwInfo = *platformDevices[0 ];
461+ hwInfo.capabilityTable .supportsImages = true ;
462+ std::unique_ptr<MockDevice> device (MockDevice::createWithNewExecutionEnvironment<MockDevice>(&hwInfo));
463+ const auto &caps = device->getDeviceInfo ();
464+
465+ EXPECT_THAT (caps.deviceExtensions , testing::HasSubstr (std::string (" cl_khr_3d_image_writes" )));
466+ }
467+
468+ TEST_F (DeviceGetCapsTest, givenNotSupportImagesWhenCapsAreCreatedThenDeviceNotReportsClKhr3dImageWritesExtensions) {
469+ HardwareInfo hwInfo = *platformDevices[0 ];
470+ hwInfo.capabilityTable .supportsImages = false ;
471+ std::unique_ptr<MockDevice> device (MockDevice::createWithNewExecutionEnvironment<MockDevice>(&hwInfo));
472+ const auto &caps = device->getDeviceInfo ();
473+ EXPECT_THAT (caps.deviceExtensions , testing::Not (testing::HasSubstr (std::string (" cl_khr_3d_image_writes" ))));
474+ }
475+
457476TEST_F (DeviceGetCapsTest, givenOpenCLVersion12WhenCapsAreCreatedThenDeviceDoesntReportClIntelSpirvExtensions) {
458477 DebugManagerStateRestore dbgRestorer;
459478 DebugManager.flags .ForceOCLVersion .set (12 );
0 commit comments