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

Change-Id: Ie35a6edb65c001b35eb9f5d2af26e765dc41c00e
Este cometimento está contido em:
Christophe Paquot criou e Christophe Paquot cometeu 2020-02-18 17:18:56 -05:00
1 ascendente e066cc6f60
cometimento b4ad4262cc
9 ficheiros modificados
+145 -120

Não existem ficheiros que correspondam à sua pesquisa

+2 -7
Ver ficheiro
@@ -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);
}