Skip to content

Commit e599d15

Browse files
wwoloszynCompute-Runtime-Automation
wwoloszyn
authored and
Compute-Runtime-Automation
committed
Use DebugManagerStateRestore for device caps tests
Change-Id: I21a0f42c468bffc69d6b2e6e641f1ae36d9baca4
1 parent b3fd643 commit e599d15

File tree

1 file changed

+32
-26
lines changed

1 file changed

+32
-26
lines changed

‎unit_tests/device/device_caps_tests.cpp

Lines changed: 32 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -364,68 +364,83 @@ TEST(Device_GetCaps, DISABLED_givenDeviceWhenCapsAreCreateThenClGLSharingIsRepor
364364
}
365365

366366
TEST(Device_GetCaps, givenOpenCLVersion21WhenCapsAreCreatedThenDeviceReportsClKhrSubgroupsExtension) {
367+
DebugManagerStateRestore dbgRestorer;
367368
DebugManager.flags.ForceOCLVersion.set(21);
368369
auto device = std::unique_ptr<Device>(MockDevice::createWithNewExecutionEnvironment<MockDevice>(platformDevices[0]));
369370
const auto &caps = device->getDeviceInfo();
370371

371372
EXPECT_THAT(caps.deviceExtensions, testing::HasSubstr(std::string("cl_khr_subgroups")));
372-
373-
DebugManager.flags.ForceOCLVersion.set(0);
374373
}
375374

376375
TEST(Device_GetCaps, givenOpenCLVersion20WhenCapsAreCreatedThenDeviceDoesntReportClKhrSubgroupsExtension) {
376+
DebugManagerStateRestore dbgRestorer;
377377
DebugManager.flags.ForceOCLVersion.set(20);
378378
auto device = std::unique_ptr<Device>(MockDevice::createWithNewExecutionEnvironment<MockDevice>(platformDevices[0]));
379379
const auto &caps = device->getDeviceInfo();
380380

381381
EXPECT_THAT(caps.deviceExtensions, testing::Not(testing::HasSubstr(std::string("cl_khr_subgroups"))));
382-
383-
DebugManager.flags.ForceOCLVersion.set(0);
384382
}
385383

386384
TEST(Device_GetCaps, givenOpenCLVersion21WhenCapsAreCreatedThenDeviceReportsClKhrIlProgramExtension) {
385+
DebugManagerStateRestore dbgRestorer;
387386
DebugManager.flags.ForceOCLVersion.set(21);
388387
auto device = std::unique_ptr<Device>(MockDevice::createWithNewExecutionEnvironment<MockDevice>(platformDevices[0]));
389388
const auto &caps = device->getDeviceInfo();
390389

391390
EXPECT_THAT(caps.deviceExtensions, testing::HasSubstr(std::string("cl_khr_il_program")));
392-
393-
DebugManager.flags.ForceOCLVersion.set(0);
394391
}
395392

396393
TEST(Device_GetCaps, givenOpenCLVersion20WhenCapsAreCreatedThenDeviceDoesntReportClKhrIlProgramExtension) {
394+
DebugManagerStateRestore dbgRestorer;
397395
DebugManager.flags.ForceOCLVersion.set(20);
398396
auto device = std::unique_ptr<Device>(MockDevice::createWithNewExecutionEnvironment<MockDevice>(platformDevices[0]));
399397
const auto &caps = device->getDeviceInfo();
400398

401399
EXPECT_THAT(caps.deviceExtensions, testing::Not(testing::HasSubstr(std::string("cl_khr_il_program"))));
402-
403-
DebugManager.flags.ForceOCLVersion.set(0);
404400
}
405401

406402
TEST(Device_GetCaps, givenEnableNV12setToTrueWhenCapsAreCreatedThenDeviceReportsNV12Extension) {
403+
DebugManagerStateRestore dbgRestorer;
407404
DebugManager.flags.EnableNV12.set(true);
408405
auto device = std::unique_ptr<Device>(MockDevice::createWithNewExecutionEnvironment<MockDevice>(platformDevices[0]));
409406
const auto &caps = device->getDeviceInfo();
410407

411408
EXPECT_THAT(caps.deviceExtensions, testing::HasSubstr(std::string("cl_intel_planar_yuv")));
412409
EXPECT_TRUE(caps.nv12Extension);
413-
414-
DebugManager.flags.EnableNV12.set(false);
415410
}
416411

417412
TEST(Device_GetCaps, givenEnablePackedYuvsetToTrueWhenCapsAreCreatedThenDeviceReportsPackedYuvExtension) {
413+
DebugManagerStateRestore dbgRestorer;
418414
DebugManager.flags.EnablePackedYuv.set(true);
419415
auto device = std::unique_ptr<Device>(MockDevice::createWithNewExecutionEnvironment<MockDevice>(platformDevices[0]));
420416
const auto &caps = device->getDeviceInfo();
421417

422418
EXPECT_THAT(caps.deviceExtensions, testing::HasSubstr(std::string("cl_intel_packed_yuv")));
423419
EXPECT_TRUE(caps.packedYuvExtension);
420+
}
421+
422+
TEST(Device_GetCaps, givenEnableNV12setToFalseWhenCapsAreCreatedThenDeviceDoesNotReportNV12Extension) {
423+
DebugManagerStateRestore dbgRestorer;
424+
DebugManager.flags.EnableNV12.set(false);
425+
auto device = std::unique_ptr<Device>(MockDevice::createWithNewExecutionEnvironment<MockDevice>(platformDevices[0]));
426+
const auto &caps = device->getDeviceInfo();
424427

428+
EXPECT_THAT(caps.deviceExtensions, testing::Not(testing::HasSubstr(std::string("cl_intel_planar_yuv"))));
429+
EXPECT_FALSE(caps.nv12Extension);
430+
}
431+
432+
TEST(Device_GetCaps, givenEnablePackedYuvsetToFalseWhenCapsAreCreatedThenDeviceDoesNotReportPackedYuvExtension) {
433+
DebugManagerStateRestore dbgRestorer;
425434
DebugManager.flags.EnablePackedYuv.set(false);
435+
auto device = std::unique_ptr<Device>(MockDevice::createWithNewExecutionEnvironment<MockDevice>(platformDevices[0]));
436+
const auto &caps = device->getDeviceInfo();
437+
438+
EXPECT_THAT(caps.deviceExtensions, testing::Not(testing::HasSubstr(std::string("cl_intel_packed_yuv"))));
439+
EXPECT_FALSE(caps.packedYuvExtension);
426440
}
427441

428442
TEST(Device_GetCaps, givenEnableVmeSetToTrueAndDeviceSupportsVmeWhenCapsAreCreatedThenDeviceReportsVmeExtensionAndBuiltins) {
443+
DebugManagerStateRestore dbgRestorer;
429444
DebugManager.flags.EnableIntelVme.set(true);
430445
auto hwInfo = *platformDevices[0];
431446
hwInfo.capabilityTable.supportsVme = true;
@@ -436,11 +451,10 @@ TEST(Device_GetCaps, givenEnableVmeSetToTrueAndDeviceSupportsVmeWhenCapsAreCreat
436451
EXPECT_TRUE(caps.vmeExtension);
437452

438453
EXPECT_THAT(caps.builtInKernels, testing::HasSubstr("block_motion_estimate_intel"));
439-
440-
DebugManager.flags.EnableIntelVme.set(false);
441454
}
442455

443456
TEST(Device_GetCaps, givenEnableVmeSetToTrueAndDeviceDoesNotSupportVmeWhenCapsAreCreatedThenDeviceDoesNotReportVmeExtensionAndBuiltins) {
457+
DebugManagerStateRestore dbgRestorer;
444458
DebugManager.flags.EnableIntelVme.set(true);
445459
auto hwInfo = *platformDevices[0];
446460
hwInfo.capabilityTable.supportsVme = false;
@@ -451,11 +465,10 @@ TEST(Device_GetCaps, givenEnableVmeSetToTrueAndDeviceDoesNotSupportVmeWhenCapsAr
451465
EXPECT_FALSE(caps.vmeExtension);
452466

453467
EXPECT_THAT(caps.builtInKernels, testing::Not(testing::HasSubstr("block_motion_estimate_intel")));
454-
455-
DebugManager.flags.EnableIntelVme.set(false);
456468
}
457469

458470
TEST(Device_GetCaps, givenEnableVmeSetToFalseAndDeviceDoesNotSupportVmeWhenCapsAreCreatedThenDeviceDoesNotReportVmeExtensionAndBuiltins) {
471+
DebugManagerStateRestore dbgRestorer;
459472
DebugManager.flags.EnableIntelVme.set(false);
460473
auto hwInfo = *platformDevices[0];
461474
hwInfo.capabilityTable.supportsVme = false;
@@ -466,11 +479,10 @@ TEST(Device_GetCaps, givenEnableVmeSetToFalseAndDeviceDoesNotSupportVmeWhenCapsA
466479
EXPECT_FALSE(caps.vmeExtension);
467480

468481
EXPECT_THAT(caps.builtInKernels, testing::Not(testing::HasSubstr("block_motion_estimate_intel")));
469-
470-
DebugManager.flags.EnableIntelVme.set(false);
471482
}
472483

473484
TEST(Device_GetCaps, givenEnableVmeSetToFalseAndDeviceSupportsVmeWhenCapsAreCreatedThenDeviceDoesNotReportVmeExtensionAndBuiltins) {
485+
DebugManagerStateRestore dbgRestorer;
474486
DebugManager.flags.EnableIntelVme.set(false);
475487
auto hwInfo = *platformDevices[0];
476488
hwInfo.capabilityTable.supportsVme = true;
@@ -481,11 +493,10 @@ TEST(Device_GetCaps, givenEnableVmeSetToFalseAndDeviceSupportsVmeWhenCapsAreCrea
481493
EXPECT_FALSE(caps.vmeExtension);
482494

483495
EXPECT_THAT(caps.builtInKernels, testing::Not(testing::HasSubstr("block_motion_estimate_intel")));
484-
485-
DebugManager.flags.EnableIntelVme.set(false);
486496
}
487497

488498
TEST(Device_GetCaps, givenEnableAdvancedVmeSetToTrueAndDeviceSupportsVmeWhenCapsAreCreatedThenDeviceReportsAdvancedVmeExtensionAndBuiltins) {
499+
DebugManagerStateRestore dbgRestorer;
489500
DebugManager.flags.EnableIntelAdvancedVme.set(true);
490501
auto hwInfo = *platformDevices[0];
491502
hwInfo.capabilityTable.supportsVme = true;
@@ -496,11 +507,10 @@ TEST(Device_GetCaps, givenEnableAdvancedVmeSetToTrueAndDeviceSupportsVmeWhenCaps
496507

497508
EXPECT_THAT(caps.builtInKernels, testing::HasSubstr("block_advanced_motion_estimate_check_intel"));
498509
EXPECT_THAT(caps.builtInKernels, testing::HasSubstr("block_advanced_motion_estimate_bidirectional_check_intel"));
499-
500-
DebugManager.flags.EnableIntelAdvancedVme.set(false);
501510
}
502511

503512
TEST(Device_GetCaps, givenEnableAdvancedVmeSetToTrueAndDeviceDoesNotSupportVmeWhenCapsAreCreatedThenDeviceDoesNotReportAdvancedVmeExtensionAndBuiltins) {
513+
DebugManagerStateRestore dbgRestorer;
504514
DebugManager.flags.EnableIntelAdvancedVme.set(true);
505515
auto hwInfo = *platformDevices[0];
506516
hwInfo.capabilityTable.supportsVme = false;
@@ -511,11 +521,10 @@ TEST(Device_GetCaps, givenEnableAdvancedVmeSetToTrueAndDeviceDoesNotSupportVmeWh
511521

512522
EXPECT_THAT(caps.builtInKernels, testing::Not(testing::HasSubstr("block_advanced_motion_estimate_check_intel")));
513523
EXPECT_THAT(caps.builtInKernels, testing::Not(testing::HasSubstr("block_advanced_motion_estimate_bidirectional_check_intel")));
514-
515-
DebugManager.flags.EnableIntelAdvancedVme.set(false);
516524
}
517525

518526
TEST(Device_GetCaps, givenEnableAdvancedVmeSetToFalseAndDeviceDoesNotSupportVmeWhenCapsAreCreatedThenDeviceDoesNotReportAdvancedVmeExtensionAndBuiltins) {
527+
DebugManagerStateRestore dbgRestorer;
519528
DebugManager.flags.EnableIntelAdvancedVme.set(false);
520529
auto hwInfo = *platformDevices[0];
521530
hwInfo.capabilityTable.supportsVme = false;
@@ -526,11 +535,10 @@ TEST(Device_GetCaps, givenEnableAdvancedVmeSetToFalseAndDeviceDoesNotSupportVmeW
526535

527536
EXPECT_THAT(caps.builtInKernels, testing::Not(testing::HasSubstr("block_advanced_motion_estimate_check_intel")));
528537
EXPECT_THAT(caps.builtInKernels, testing::Not(testing::HasSubstr("block_advanced_motion_estimate_bidirectional_check_intel")));
529-
530-
DebugManager.flags.EnableIntelAdvancedVme.set(false);
531538
}
532539

533540
TEST(Device_GetCaps, givenEnableAdvancedVmeSetToFalseAndDeviceSupportsVmeWhenCapsAreCreatedThenDeviceDoesNotReportAdvancedVmeExtensionAndBuiltins) {
541+
DebugManagerStateRestore dbgRestorer;
534542
DebugManager.flags.EnableIntelAdvancedVme.set(false);
535543
auto hwInfo = *platformDevices[0];
536544
hwInfo.capabilityTable.supportsVme = true;
@@ -541,8 +549,6 @@ TEST(Device_GetCaps, givenEnableAdvancedVmeSetToFalseAndDeviceSupportsVmeWhenCap
541549

542550
EXPECT_THAT(caps.builtInKernels, testing::Not(testing::HasSubstr("block_advanced_motion_estimate_check_intel")));
543551
EXPECT_THAT(caps.builtInKernels, testing::Not(testing::HasSubstr("block_advanced_motion_estimate_bidirectional_check_intel")));
544-
545-
DebugManager.flags.EnableIntelAdvancedVme.set(false);
546552
}
547553

548554
TEST(Device_GetCaps, byDefaultVmeIsTurnedOn) {

0 commit comments

Comments
 (0)