File torchvision-fix-build.patch of Package python-torchvision-stable
diff --git a/setup.py b/setup.py
index ce1cd90..bf5bcc5 100644
--- a/setup.py
+++ b/setup.py
@@ -174,7 +174,7 @@ def get_extensions():
hipify_python.hipify(
project_directory=this_dir,
output_directory=this_dir,
- includes="torchvision/csrc/ops/cuda/*",
+ includes=["*/torchvision/csrc/ops/cuda/*"],
show_detailed=True,
is_pytorch_extension=True,
)
@@ -223,8 +223,8 @@ def get_extensions():
extra_compile_args["nvcc"].append("-O0")
extra_compile_args["nvcc"].append("-g")
else:
- print("Compiling with debug mode OFF")
- extra_compile_args["cxx"].append("-g0")
+ print("Compiling release with debug symbols")
+ extra_compile_args["cxx"].append("-g")
sources = [os.path.join(extensions_dir, s) for s in sources]
@@ -380,8 +380,8 @@ def get_extensions():
ffmpeg_bin = os.path.dirname(ffmpeg_exe)
ffmpeg_root = os.path.dirname(ffmpeg_bin)
- ffmpeg_include_dir = os.path.join(ffmpeg_root, "include")
- ffmpeg_library_dir = os.path.join(ffmpeg_root, "lib")
+ ffmpeg_include_dir = os.path.join(ffmpeg_root, "include/ffmpeg")
+ ffmpeg_library_dir = os.path.join(ffmpeg_root, "lib64")
gcc = os.environ.get("CC", shutil.which("gcc"))
platform_tag = subprocess.run([gcc, "-print-multiarch"], stdout=subprocess.PIPE)