diff --git a/rocclr/runtime/device/rocm/rocdefs.hpp b/rocclr/runtime/device/rocm/rocdefs.hpp index eb12a99c83..b08349f1e7 100644 --- a/rocclr/runtime/device/rocm/rocdefs.hpp +++ b/rocclr/runtime/device/rocm/rocdefs.hpp @@ -31,6 +31,7 @@ const HsaDeviceId HSA_FIJI_ID = 5; const HsaDeviceId HSA_HAWAII_ID = 6; const HsaDeviceId HSA_ELLESMERE_ID = 7; const HsaDeviceId HSA_BAFFIN_ID = 8; +const HsaDeviceId HSA_VEGA10_ID = 9; const HsaDeviceId HSA_INVALID_DEVICE_ID = -1; static const AMDDeviceInfo DeviceInfo[] = { @@ -43,7 +44,8 @@ static const AMDDeviceInfo DeviceInfo[] = { /* TARGET_FIJI */ {HSA_FIJI_ID, "", "fiji", "Fiji", 4, 16, 1, 256, 64 * Ki, 32, 0, 0 }, /* TARGET HAWAII */ {HSA_HAWAII_ID, "", "hawaii", "Hawaii", 4, 16, 1, 256, 64 * Ki, 32, 0, 0 }, /* TARGET ELLESMERE */ {HSA_ELLESMERE_ID, "", "polaris10", "Ellesmere", 4, 16, 1, 256, 64 * Ki, 32, 0, 0 }, - /* TARGET BAFFIN */ {HSA_BAFFIN_ID, "", "polaris11", "Baffin", 4, 16, 1, 256, 64 * Ki, 32, 0, 0 } + /* TARGET BAFFIN */ {HSA_BAFFIN_ID, "", "polaris11", "Baffin", 4, 16, 1, 256, 64 * Ki, 32, 0, 0 }, + /* TARGET VEGA10 */ {HSA_VEGA10_ID, "", "gfx900", "gfx900", 4, 16, 1, 256, 64 * Ki, 32, 0, 0 } }; } diff --git a/rocclr/runtime/device/rocm/rocdevice.cpp b/rocclr/runtime/device/rocm/rocdevice.cpp index 7d462eddfa..ca33e9cead 100644 --- a/rocclr/runtime/device/rocm/rocdevice.cpp +++ b/rocclr/runtime/device/rocm/rocdevice.cpp @@ -29,6 +29,7 @@ #include "vi_id.h" #include "cz_id.h" #include "ci_id.h" +#include "ai_id.h" #include #include #include @@ -147,6 +148,8 @@ static HsaDeviceId getHsaDeviceId(hsa_agent_t device, uint32_t& pci_id) { case DEVICE_ID_VI_BAFFIN_M_67E7: case DEVICE_ID_VI_BAFFIN_M_67E9: return HSA_BAFFIN_ID; + case DEVICE_ID_AI_GREENLAND_P_687F: + return HSA_VEGA10_ID; default: return HSA_INVALID_DEVICE_ID; }