retain pi_device when reusing sub-device

bso-intel · bso-intel · commit 040d81ba0859 · 2021-03-10T11:05:48.000-08:00
Signed-off-by: Byoungro So &lt;byoungro.so@intel.com&gt;
diff --git a/sycl/plugins/level_zero/pi_level_zero.cpp b/sycl/plugins/level_zero/pi_level_zero.cpp
@@ -1809,6 +1809,8 @@ pi_result piDevicePartition(pi_device Device,
   if (OutDevices && NumDevices <= Device->SubDevices.size()) {
     for (uint32_t I = 0; I < NumDevices; I++) {
       OutDevices[I] = Device->SubDevices[I];
+      // reusing the same pi_device needs to increment the reference count
+      piDeviceRetain(OutDevices[I]);
     }
     if (OutNumDevices)
       *OutNumDevices = NumDevices;

Original file line number	Diff line number	Diff line change
`@@ -1809,6 +1809,8 @@ pi_result piDevicePartition(pi_device Device,`
`1809`	`1809`	`if (OutDevices && NumDevices <= Device->SubDevices.size()) {`
`1810`	`1810`	`for (uint32_t I = 0; I < NumDevices; I++) {`
`1811`	`1811`	`OutDevices[I] = Device->SubDevices[I];`
	`1812`	`+ // reusing the same pi_device needs to increment the reference count`
	`1813`	`+ piDeviceRetain(OutDevices[I]);`
`1812`	`1814`	`}`
`1813`	`1815`	`if (OutNumDevices)`
`1814`	`1816`	`*OutNumDevices = NumDevices;`