Introducing hip::Device which wraps around amd::Context and deviceId

Change-Id: Ie35a6edb65c001b35eb9f5d2af26e765dc41c00e
Αυτή η υποβολή περιλαμβάνεται σε:
Christophe Paquot authored and Christophe Paquot committed 2020-02-18 17:18:56 -05:00
1 γονέας e066cc6f60
υποβολή b4ad4262cc
9 files changed
+145 -120

No files matched your search

@@ -430,12 +430,7 @@ hipError_t hipDeviceSynchronize ( void ) {
}
int ihipGetDevice() {
for (unsigned int i = 0; i < g_devices.size(); i++) {
if (g_devices[i] == hip::getCurrentContext()) {
return i;
}
}
return -1;
return hip::getCurrentDevice()->deviceId();
}
hipError_t hipGetDevice ( int* deviceId ) {
@@ -483,7 +478,7 @@ hipError_t hipSetDevice ( int device ) {
HIP_INIT_API(hipSetDevice, device);
if (static_cast<unsigned int>(device) < g_devices.size()) {
hip::setCurrentContext(device);
hip::setCurrentDevice(device);
HIP_RETURN(hipSuccess);
}