[SYCL] Fix subdevices caching. (#3577)

rdeodhar · web-flow · commit 4c34f93af1bf · 2021-04-21T08:07:30.000+03:00
This change corrects an error in the caching of subdevices.

Signed-off-by: rdeodhar &lt;rajiv.deodhar@intel.com&gt;
diff --git a/sycl/plugins/level_zero/pi_level_zero.cpp b/sycl/plugins/level_zero/pi_level_zero.cpp
@@ -2027,9 +2027,9 @@ pi_result piDevicePartition(pi_device Device,
           return Result;
         }
         OutDevices[I] = PiSubDevice.get();
-        Platform->PiDevicesCache.push_back(std::move(PiSubDevice));
         // save pointers to sub-devices for quick retrieval in the future.
-        Device->SubDevices.push_back(Dev);
+        Device->SubDevices.push_back(PiSubDevice.get());
+        Platform->PiDevicesCache.push_back(std::move(PiSubDevice));
       }
     }
     delete[] ZeSubdevices;

Original file line number	Diff line number	Diff line change
`@@ -2027,9 +2027,9 @@ pi_result piDevicePartition(pi_device Device,`
`2027`	`2027`	`return Result;`
`2028`	`2028`	`}`
`2029`	`2029`	`OutDevices[I] = PiSubDevice.get();`
`2030`		`- Platform->PiDevicesCache.push_back(std::move(PiSubDevice));`
`2031`	`2030`	`// save pointers to sub-devices for quick retrieval in the future.`
`2032`		`- Device->SubDevices.push_back(Dev);`
	`2031`	`+ Device->SubDevices.push_back(PiSubDevice.get());`
	`2032`	`+ Platform->PiDevicesCache.push_back(std::move(PiSubDevice));`
`2033`	`2033`	`}`
`2034`	`2034`	`}`
`2035`	`2035`	`delete[] ZeSubdevices;`