commit:     14fc580a6158a3b1d3555c38cf00bd7dbb4e4d8f
Author:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
AuthorDate: Sat Feb 22 16:01:41 2025 +0000
Commit:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
CommitDate: Sat Feb 22 16:02:08 2025 +0000
URL:        https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=14fc580a

sci-libs/torchvision: better fix for cuda build

Signed-off-by: Alfredo Tupone <tupone <AT> gentoo.org>

 .../torchvision/files/torchvision-0.20.0-gentoo.patch    | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/sci-libs/torchvision/files/torchvision-0.20.0-gentoo.patch 
b/sci-libs/torchvision/files/torchvision-0.20.0-gentoo.patch
index a622e41ec1ea..2633660cafa5 100644
--- a/sci-libs/torchvision/files/torchvision-0.20.0-gentoo.patch
+++ b/sci-libs/torchvision/files/torchvision-0.20.0-gentoo.patch
@@ -1,11 +1,19 @@
---- a/setup.py 2025-02-21 22:49:07.255937371 +0100
-+++ b/setup.py 2025-02-21 22:49:20.372975254 +0100
-@@ -127,7 +127,7 @@
+--- a/setup.py 2025-02-22 16:44:42.102439182 +0100
++++ b/setup.py 2025-02-22 16:46:52.393841294 +0100
+@@ -6,6 +6,7 @@
+ import subprocess
+ import sys
+ import warnings
++import shlex
+ from pathlib import Path
+ 
+ import torch
+@@ -127,7 +128,7 @@
              if NVCC_FLAGS is None:
                  nvcc_flags = []
              else:
 -                nvcc_flags = nvcc_flags.split(" ")
-+                nvcc_flags = NVCC_FLAGS.split(" ")
++                nvcc_flags = shlex.split(NVCC_FLAGS)
          extra_compile_args["nvcc"] = nvcc_flags
  
      if sys.platform == "win32":

Reply via email to