Bazel: bugfix for libtorch_gpu.BUILD

c5abb9fa · storypku · Liu Jiaming · d9c5ce95 · c5abb9fa
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

third_party/libtorch/libtorch_gpu.BUILD third_party/libtorch/libtorch_gpu.BUILD +2 -1

未找到文件。
--- a/third_party/libtorch/libtorch_gpu.BUILD
+++ b/third_party/libtorch/libtorch_gpu.BUILD
@@ -19,7 +19,8 @@ cc_library(
    ],
    linkstatic = False,
    deps = [
+        "@local_config_cuda//cuda:cudart",
+        "@local_config_python//:python_headers",
        "@local_config_python//:python_lib",
-        "@local_config_cuda//:cudart",
    ],
 )