rocm: update rocm gpu targets

zml · Sep 9, 2024 · 03d3d81 · 03d3d81
1 parent 0159000
commit 03d3d81
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/rocm/BUILD.bazel b/rocm/BUILD.bazel
@@ -83,7 +83,7 @@ oci_image(
         "PATH": "/usr/local/bin:/opt/rocm/opencl/bin:/opt/rocm/bin:/opt/rocm/hcc/bin:/opt/rocm/hip/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin",
         "ROCM_PATH": "/opt/rocm-6.2.0/",
         "TERM": "xterm",
-        "TF_ROCM_AMDGPU_TARGETS": "gfx900\\ngfx906\\ngfx908",
+        "TF_ROCM_AMDGPU_TARGETS": "gfx900,gfx906,gfx908,gfx90a,gfx940,gfx941,gfx942,gfx1030,gfx1100",
     },
 )
 

diff --git a/rocm/targets.lst b/rocm/targets.lst
@@ -1,3 +1,9 @@
 gfx900
 gfx906
-gfx908
+gfx908
+gfx90a
+gfx940
+gfx941
+gfx942
+gfx1030
+gfx1100
-Original file line number
+Diff line change
@@ -1,3 +1,9 @@
     gfx900
     gfx906
-    gfx908
+    gfx908
+    gfx90a
+    gfx940
+    gfx941
+    gfx942
+    gfx1030
+    gfx1100