Improve queries with number of SubDevices

Signed-off-by: Bartosz Dunajski <bartosz.dunajski@intel.com>
This commit is contained in:
Bartosz Dunajski
2021-08-31 16:49:46 +00:00
committed by Compute-Runtime-Automation
parent 69ae9dc9c2
commit 0345d9f707
44 changed files with 229 additions and 173 deletions

View File

@@ -63,11 +63,11 @@ TEST(ExecutionEnvironment, WhenCreatingDevicesThenThoseDevicesAddRefcountsToExec
auto expectedRefCounts = executionEnvironment->getRefInternalCount();
auto devices = DeviceFactory::createDevices(*executionEnvironment);
EXPECT_LT(0u, devices[0]->getNumAvailableDevices());
if (devices[0]->getNumAvailableDevices() > 1) {
EXPECT_LE(0u, devices[0]->getNumSubDevices());
if (devices[0]->getNumSubDevices() > 1) {
expectedRefCounts++;
}
expectedRefCounts += devices[0]->getNumAvailableDevices();
expectedRefCounts += std::max(devices[0]->getNumSubDevices(), 1u);
EXPECT_EQ(expectedRefCounts, executionEnvironment->getRefInternalCount());
}