@@ -93,8 +93,8 @@ EXTERN int omp_get_device_num(void) {
9393 return HostDevice;
9494}
9595
96- static inline bool is_host_device_uid (const char *DeviceUid) {
97- return strcmp (DeviceUid, GenericPluginTy::getHostDeviceUid ()) == 0 ;
96+ static inline bool is_initial_device_uid (const char *DeviceUid) {
97+ return strcmp (DeviceUid, GenericPluginTy::getInitialDeviceUid ()) == 0 ;
9898}
9999
100100EXTERN int omp_get_device_from_uid (const char *DeviceUid) {
@@ -105,8 +105,8 @@ EXTERN int omp_get_device_from_uid(const char *DeviceUid) {
105105 DP (" Call to omp_get_device_from_uid returning omp_invalid_device\n " );
106106 return omp_invalid_device;
107107 }
108- if (is_host_device_uid (DeviceUid)) {
109- DP (" Call to omp_get_device_from_uid returning host device number %d\n " ,
108+ if (is_initial_device_uid (DeviceUid)) {
109+ DP (" Call to omp_get_device_from_uid returning initial device number %d\n " ,
110110 omp_get_initial_device ());
111111 return omp_get_initial_device ();
112112 }
@@ -135,17 +135,15 @@ EXTERN const char *omp_get_uid_from_device(int DeviceNum) {
135135 return nullptr ;
136136 }
137137 if (DeviceNum == omp_get_initial_device ()) {
138- DP (" Call to omp_get_uid_from_device returning host device UID\n " );
139- return GenericPluginTy::getHostDeviceUid ();
138+ DP (" Call to omp_get_uid_from_device returning initial device UID\n " );
139+ return GenericPluginTy::getInitialDeviceUid ();
140140 }
141141
142- llvm::Expected<DeviceTy &> Device = PM->getDevice (DeviceNum);
143- if (!Device) {
144- FATAL_MESSAGE (DeviceNum, " %s" , toString (Device.takeError ()).c_str ());
145- return nullptr ;
146- }
142+ auto DeviceOrErr = PM->getDevice (DeviceNum);
143+ if (!DeviceOrErr)
144+ FATAL_MESSAGE (DeviceNum, " %s" , toString (DeviceOrErr.takeError ()).c_str ());
147145
148- const char *Uid = Device ->RTL ->getDeviceUid (Device ->RTLDeviceID );
146+ const char *Uid = DeviceOrErr ->RTL ->getDeviceUid (DeviceOrErr ->RTLDeviceID );
149147 DP (" Call to omp_get_uid_from_device returning %s\n " , Uid);
150148 return Uid;
151149}
0 commit comments