@@ -70,8 +70,8 @@ EXTERN int omp_get_device_num(void) {
7070 return HostDevice;
7171}
7272
73- static inline bool is_host_device_uid (const char *DeviceUid) {
74- return strcmp (DeviceUid, GenericPluginTy::getHostDeviceUid ()) == 0 ;
73+ static inline bool is_initial_device_uid (const char *DeviceUid) {
74+ return strcmp (DeviceUid, GenericPluginTy::getInitialDeviceUid ()) == 0 ;
7575}
7676
7777EXTERN int omp_get_device_from_uid (const char *DeviceUid) {
@@ -82,8 +82,8 @@ EXTERN int omp_get_device_from_uid(const char *DeviceUid) {
8282 DP (" Call to omp_get_device_from_uid returning omp_invalid_device\n " );
8383 return omp_invalid_device;
8484 }
85- if (is_host_device_uid (DeviceUid)) {
86- DP (" Call to omp_get_device_from_uid returning host device number %d\n " ,
85+ if (is_initial_device_uid (DeviceUid)) {
86+ DP (" Call to omp_get_device_from_uid returning initial device number %d\n " ,
8787 omp_get_initial_device ());
8888 return omp_get_initial_device ();
8989 }
@@ -112,17 +112,15 @@ EXTERN const char *omp_get_uid_from_device(int DeviceNum) {
112112 return nullptr ;
113113 }
114114 if (DeviceNum == omp_get_initial_device ()) {
115- DP (" Call to omp_get_uid_from_device returning host device UID\n " );
116- return GenericPluginTy::getHostDeviceUid ();
115+ DP (" Call to omp_get_uid_from_device returning initial device UID\n " );
116+ return GenericPluginTy::getInitialDeviceUid ();
117117 }
118118
119- llvm::Expected<DeviceTy &> Device = PM->getDevice (DeviceNum);
120- if (!Device) {
121- FATAL_MESSAGE (DeviceNum, " %s" , toString (Device.takeError ()).c_str ());
122- return nullptr ;
123- }
119+ auto DeviceOrErr = PM->getDevice (DeviceNum);
120+ if (!DeviceOrErr)
121+ FATAL_MESSAGE (DeviceNum, " %s" , toString (DeviceOrErr.takeError ()).c_str ());
124122
125- const char *Uid = Device ->RTL ->getDeviceUid (Device ->RTLDeviceID );
123+ const char *Uid = DeviceOrErr ->RTL ->getDeviceUid (DeviceOrErr ->RTLDeviceID );
126124 DP (" Call to omp_get_uid_from_device returning %s\n " , Uid);
127125 return Uid;
128126}
0 commit comments