From 59eb31c2fbc7e90a7f5bb69ed6864fe2980521fe Mon Sep 17 00:00:00 2001 From: charlie Date: Tue, 19 Nov 2024 11:10:30 -0600 Subject: [PATCH] Change gfx number checking for fp8 hw support --- src/targets/gpu/device_name.cpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/targets/gpu/device_name.cpp b/src/targets/gpu/device_name.cpp index e3f47a4f3b0..d4ebab89695 100644 --- a/src/targets/gpu/device_name.cpp +++ b/src/targets/gpu/device_name.cpp @@ -52,13 +52,15 @@ std::string get_device_name() bool gfx_has_fp8fnuz_intrinsics() { const auto device_name = trim(split_string(get_device_name(), ':').front()); - return (starts_with(device_name, "gfx9") and device_name >= "gfx940"); + return (starts_with(device_name, "gfx94")); } bool gfx_has_fp8ocp_intrinsics() { const auto device_name = trim(split_string(get_device_name(), ':').front()); - return (starts_with(device_name, "gfx12") and device_name >= "gfx1200"); + bool is_navi4x = starts_with(device_name, "gfx12") and device_name >= "gfx1200"; + bool is_mi35x = starts_with(device_name, "gfx9") and device_name >= "gfx950"; + return (is_navi4x or is_mi35x); } } // namespace gpu