[OpenMP][libomptarget] Init device when printing device info

kevinsala · kevinsala · commit aa326559c443 · 2023-05-09T18:47:46.000+02:00
This patch fixes the printing of device information. Devices are initialized before printing its information. Fixes #61392 Differential Revision: https://reviews.llvm.org/D146081
diff --git a/openmp/libomptarget/src/interface.cpp b/openmp/libomptarget/src/interface.cpp
@@ -384,6 +384,12 @@ EXTERN void __tgt_set_info_flag(uint32_t NewInfoLevel) {
 }
 
 EXTERN int __tgt_print_device_info(int64_t DeviceId) {
+  // Make sure the device is ready.
+  if (!deviceIsReady(DeviceId)) {
+    DP("Device %" PRId64 " is not ready\n", DeviceId);
+    return OMP_TGT_FAIL;
+  }
+
   return PM->Devices[DeviceId]->printDeviceInfo(
       PM->Devices[DeviceId]->RTLDeviceID);
 }

Original file line number	Diff line number	Diff line change
`@@ -384,6 +384,12 @@ EXTERN void __tgt_set_info_flag(uint32_t NewInfoLevel) {`
`384`	`384`	`}`
`385`	`385`
`386`	`386`	`EXTERN int __tgt_print_device_info(int64_t DeviceId) {`
	`387`	`+ // Make sure the device is ready.`
	`388`	`+ if (!deviceIsReady(DeviceId)) {`
	`389`	`+ DP("Device %" PRId64 " is not ready\n", DeviceId);`
	`390`	`+ return OMP_TGT_FAIL;`
	`391`	`+ }`
	`392`	`+`
`387`	`393`	`return PM->Devices[DeviceId]->printDeviceInfo(`
`388`	`394`	`PM->Devices[DeviceId]->RTLDeviceID);`
`389`	`395`	`}`