Fix device count assignment based on zeDrivers array

nrspruit · nrspruit · commit 0352fe0a4faf · 2024-11-22T15:55:52.000-08:00
Signed-off-by: Neil R. Spruit &lt;neil.r.spruit@intel.com&gt;
diff --git a/source/adapters/level_zero/adapter.cpp b/source/adapters/level_zero/adapter.cpp
@@ -157,10 +157,11 @@ ur_result_t initPlatforms(PlatformVec &platforms,
       }
     }
   } else {
-    ZeDriverCount = ZeDriverGetCount;
-    ZeDrivers.resize(ZeDriverCount);
+    ZeDrivers.resize(ZeDriverGetCount);
     ZeDrivers.assign(ZeDriverGetHandles.begin(), ZeDriverGetHandles.end());
   }
+  ZeDriverCount = ZeDrivers.size();
+  logger::debug("\n{} L0 Drivers found.\n", ZeDriverCount);
   for (uint32_t I = 0; I < ZeDriverCount; ++I) {
     // Keep track of the first platform init for this Driver
     bool DriverPlatformInit = false;

Original file line number	Diff line number	Diff line change
`@@ -157,10 +157,11 @@ ur_result_t initPlatforms(PlatformVec &platforms,`
`157`	`157`	`}`
`158`	`158`	`}`
`159`	`159`	`} else {`
`160`		`- ZeDriverCount = ZeDriverGetCount;`
`161`		`- ZeDrivers.resize(ZeDriverCount);`
	`160`	`+ ZeDrivers.resize(ZeDriverGetCount);`
`162`	`161`	`ZeDrivers.assign(ZeDriverGetHandles.begin(), ZeDriverGetHandles.end());`
`163`	`162`	`}`
	`163`	`+ ZeDriverCount = ZeDrivers.size();`
	`164`	`+ logger::debug("\n{} L0 Drivers found.\n", ZeDriverCount);`
`164`	`165`	`for (uint32_t I = 0; I < ZeDriverCount; ++I) {`
`165`	`166`	`// Keep track of the first platform init for this Driver`
`166`	`167`	`bool DriverPlatformInit = false;`