[ROCm/rocjpeg commit: 9f41913587]
Этот коммит содержится в:
Aryan Salmanpour
2025-01-09 10:38:29 -05:00
коммит произвёл GitHub
родитель 2cd9e6a129
Коммит a50863d456
+3 -11
Просмотреть файл
@@ -1006,19 +1006,11 @@ void RocJpegVappiDecoder::GetDrmNodeOffset(std::string device_name, uint8_t devi
break;
case kCpx:
// Note: The MI300 series share the same gfx_arch_name (gfx942).
// Therefore, we cannot use gfx942 to distinguish between MI300A, MI308, etc.
// Instead, use the device name to identify MI300A, MI308, etc.
// Therefore, we cannot use gfx942 to distinguish between MI300X, MI300A etc.
// Instead, use the device name to identify MI300A, etc.
std::string mi300a = "MI300A";
size_t found_mi300a = device_name.find(mi300a);
std::string mi308 = "MI308";
size_t found_mi308 = device_name.find(mi308);
if (found_mi308 != std::string::npos) {
if (device_id < visible_devices.size()) {
offset = (visible_devices[device_id] % 4);
} else {
offset = (device_id % 4);
}
} else if (found_mi300a != std::string::npos) {
if (found_mi300a != std::string::npos) {
if (device_id < visible_devices.size()) {
offset = (visible_devices[device_id] % 6);
} else {