Просмотр исходного кода

libtorch: remove nvtx dependency since v2.4.1 (#6158)

icysky 8 месяцев назад
Родитель
Сommit
71b100e8f6
1 измененных файлов с 3 добавлено и 1 удалено
  1. 3 1
      packages/l/libtorch/xmake.lua

+ 3 - 1
packages/l/libtorch/xmake.lua

@@ -62,7 +62,9 @@ package("libtorch")
         end
         end
         if package:config("cuda") then
         if package:config("cuda") then
             package:add("deps", "cuda", {configs = {utils = {"nvrtc", "cudnn", "cufft", "curand", "cublas", "cudart_static"}}})
             package:add("deps", "cuda", {configs = {utils = {"nvrtc", "cudnn", "cufft", "curand", "cublas", "cudart_static"}}})
-            package:add("deps", "nvtx")
+            if package:version():lt("2.5.0") then
+                package:add("deps", "nvtx")
+            end
         end
         end
         if package:config("distributed") then
         if package:config("distributed") then
             package:add("deps", "libuv")
             package:add("deps", "libuv")