Skip to content

Commit 604dd77

Browse files
committed
adapt to offload API
1 parent c25a941 commit 604dd77

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
@@ -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

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

Comments
 (0)