Add missing dependency

ROCm · Nov 12, 2024 · ae875cd · ae875cd
1 parent b5c309e
commit ae875cd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/xla/service/gpu/kernels/BUILD b/xla/service/gpu/kernels/BUILD
@@ -174,7 +174,9 @@ gpu_kernel_library(
         "//xla:types",
         "//xla/stream_executor/gpu:gpu_types_header",
         "@tsl//tsl/lib/math:math_util",
-    ],
+    ] + if_rocm_is_configured([
+        "@local_config_rocm//rocm:rocm",
+    ]),
 )
 
 xla_test(