Skip to content

Commit eed9b95

Browse files
committed
adapt to offload API
1 parent 69e61df commit eed9b95

File tree

1 file changed

+10
-12
lines changed

1 file changed

+10
-12
lines changed

offload/libomptarget/OpenMP/API.cpp

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -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

7777
EXTERN 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

Comments
 (0)