FP8OCP GFX update (#3635)

ROCm · Nov 20, 2024 · 07010b6 · 07010b6
1 parent 333161f
commit 07010b6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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_navi_with_fp8ocp = starts_with(device_name, "gfx12") and device_name >= "gfx1200";
+    bool is_mi_with_fp8ocp   = starts_with(device_name, "gfx9") and device_name >= "gfx950";
+    return (is_navi_with_fp8ocp or is_mi_with_fp8ocp);
 }
 
 } // namespace gpu