register gfx12xx

ROCm · Jan 9, 2025 · 17be9ac · 17be9ac
1 parent f55fc88
commit 17be9ac
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/xla/stream_executor/device_description.h b/xla/stream_executor/device_description.h
@@ -202,6 +202,10 @@ class RocmComputeCapability {
 
   bool gfx11_rx7900() const { return gfx_version() == "gfx1100"; }
 
+  bool gfx1200() const { return gfx_version() == "gfx1200"; }
+
+  bool gfx1201() const { return gfx_version() == "gfx1201"; }
+
   bool has_nhwc_layout_support() const { return gfx9_mi100_or_later(); }
 
   bool has_bf16_dtype_support() const { return gfx9_mi100_or_later(); }
@@ -227,9 +231,9 @@ class RocmComputeCapability {
     return gfx_version() != "gfx900" && gfx_version() != "gfx906";
   }
 
-  bool has_hipblaslt() const { return gfx9_mi200_or_later(); }
+  bool has_hipblaslt() const { return gfx9_mi200_or_later() || gfx1200() || gfx1201(); }
 
-  bool has_fp8_support() const { return gfx9_mi300(); }
+  bool has_fp8_support() const { return gfx9_mi300() || gfx1200() || gfx1201(); }
 
   RocmComputeCapabilityProto ToProto() const {
     RocmComputeCapabilityProto proto;
@@ -251,7 +255,9 @@ class RocmComputeCapability {
       "gfx90a",                       // MI200
       "gfx940",  "gfx941", "gfx942",  // MI300
       "gfx1030",                      // RX68xx / RX69xx
-      "gfx1100"                       // RX7900
+      "gfx1100",                      // RX7900
+      "gfx1200",
+      "gfx1201",
   };
 };