Enable pjrt_c_api_gpu_test

ROCm · Sep 25, 2024 · f40daa8 · f40daa8
1 parent cd4b04d
commit f40daa8
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/third_party/xla/xla/pjrt/c/BUILD b/third_party/xla/xla/pjrt/c/BUILD
@@ -361,9 +361,7 @@ xla_test(
     name = "pjrt_c_api_gpu_test",
     srcs = ["pjrt_c_api_gpu_test.cc"],
     backends = ["gpu"],
-    tags = [
-        "no_rocm", #TODO(rocm): TEMP, sync 24-06-03
-    ] + if_google([
+    tags = if_google([
         "config-cuda-only",
     ]),
     deps = [