File groundingdino-py-fix-setup.patch of Package python-groundingdino-py
--- a/setup.py 2024-08-12 01:52:02.000000000 +0200
+++ b/setup.py.mod 2025-08-06 16:32:47.578630222 +0200
@@ -20,37 +20,37 @@
# https://github.com/Oneflow-Inc/libai/blob/main/setup.py
# ------------------------------------------------------------------------------------------------
-import glob
+from setuptools import find_packages, setup
import os
-import subprocess
-
-import subprocess
-import sys
+# import subprocess
-def install_torch():
- try:
- import torch
- except ImportError:
- subprocess.check_call([sys.executable, "-m", "pip", "install", "torch"])
+# import subprocess
+# import sys
-# Call the function to ensure torch is installed
-install_torch()
-
-import torch
-from setuptools import find_packages, setup
-from torch.utils.cpp_extension import CUDA_HOME, CppExtension, CUDAExtension
+# def install_torch():
+# try:
+# import torch
+# except ImportError:
+# subprocess.check_call([sys.executable, "-m", "pip", "install", "torch"])
+#
+# # Call the function to ensure torch is installed
+# install_torch()
+#
+# import torch
+# from setuptools import find_packages, setup
+# from torch.utils.cpp_extension import CUDA_HOME, CppExtension, CUDAExtension
# groundingdino version info
-version = "0.1.0"
+version = "@VERSION@"
package_name = "groundingdino"
cwd = os.path.dirname(os.path.abspath(__file__))
-sha = "Unknown"
-try:
- sha = subprocess.check_output(["git", "rev-parse", "HEAD"], cwd=cwd).decode("ascii").strip()
-except Exception:
- pass
+# sha = "Unknown"
+# try:
+# sha = subprocess.check_output(["git", "rev-parse", "HEAD"], cwd=cwd).decode("ascii").strip()
+# except Exception:
+# pass
def write_version_file():
@@ -61,58 +61,58 @@
requirements = ["torch", "torchvision"]
-
-torch_ver = [int(x) for x in torch.__version__.split(".")[:2]]
-
-
-def get_extensions():
- this_dir = os.path.dirname(os.path.abspath(__file__))
- extensions_dir = os.path.join(this_dir, "groundingdino", "models", "GroundingDINO", "csrc")
-
- main_source = os.path.join(extensions_dir, "vision.cpp")
- sources = glob.glob(os.path.join(extensions_dir, "**", "*.cpp"))
- source_cuda = glob.glob(os.path.join(extensions_dir, "**", "*.cu")) + glob.glob(
- os.path.join(extensions_dir, "*.cu")
- )
-
- sources = [main_source] + sources
-
- extension = CppExtension
-
- extra_compile_args = {"cxx": []}
- define_macros = []
-
- if CUDA_HOME is not None and (torch.cuda.is_available() or "TORCH_CUDA_ARCH_LIST" in os.environ):
- print("Compiling with CUDA")
- extension = CUDAExtension
- sources += source_cuda
- define_macros += [("WITH_CUDA", None)]
- extra_compile_args["nvcc"] = [
- "-DCUDA_HAS_FP16=1",
- "-D__CUDA_NO_HALF_OPERATORS__",
- "-D__CUDA_NO_HALF_CONVERSIONS__",
- "-D__CUDA_NO_HALF2_OPERATORS__",
- ]
- else:
- print("Compiling without CUDA")
- define_macros += [("WITH_HIP", None)]
- extra_compile_args["nvcc"] = []
- return None
-
- sources = [os.path.join(extensions_dir, s) for s in sources]
- include_dirs = [extensions_dir]
-
- ext_modules = [
- extension(
- "groundingdino._C",
- sources,
- include_dirs=include_dirs,
- define_macros=define_macros,
- extra_compile_args=extra_compile_args,
- )
- ]
-
- return ext_modules
+#
+# torch_ver = [int(x) for x in torch.__version__.split(".")[:2]]
+#
+#
+# def get_extensions():
+# this_dir = os.path.dirname(os.path.abspath(__file__))
+# extensions_dir = os.path.join(this_dir, "groundingdino", "models", "GroundingDINO", "csrc")
+#
+# main_source = os.path.join(extensions_dir, "vision.cpp")
+# sources = glob.glob(os.path.join(extensions_dir, "**", "*.cpp"))
+# source_cuda = glob.glob(os.path.join(extensions_dir, "**", "*.cu")) + glob.glob(
+# os.path.join(extensions_dir, "*.cu")
+# )
+#
+# sources = [main_source] + sources
+#
+# extension = CppExtension
+#
+# extra_compile_args = {"cxx": []}
+# define_macros = []
+#
+# if CUDA_HOME is not None and (torch.cuda.is_available() or "TORCH_CUDA_ARCH_LIST" in os.environ):
+# print("Compiling with CUDA")
+# extension = CUDAExtension
+# sources += source_cuda
+# define_macros += [("WITH_CUDA", None)]
+# extra_compile_args["nvcc"] = [
+# "-DCUDA_HAS_FP16=1",
+# "-D__CUDA_NO_HALF_OPERATORS__",
+# "-D__CUDA_NO_HALF_CONVERSIONS__",
+# "-D__CUDA_NO_HALF2_OPERATORS__",
+# ]
+# else:
+# print("Compiling without CUDA")
+# define_macros += [("WITH_HIP", None)]
+# extra_compile_args["nvcc"] = []
+# return None
+#
+# sources = [os.path.join(extensions_dir, s) for s in sources]
+# include_dirs = [extensions_dir]
+#
+# ext_modules = [
+# extension(
+# "groundingdino._C",
+# sources,
+# include_dirs=include_dirs,
+# define_macros=define_macros,
+# extra_compile_args=extra_compile_args,
+# )
+# ]
+#
+# return ext_modules
def parse_requirements(fname="requirements.txt", with_version=True):
@@ -203,7 +203,7 @@
setup(
name="groundingdino",
- version="0.1.0",
+ version=version,
author="International Digital Economy Academy, Shilong Liu",
url="https://github.com/IDEA-Research/GroundingDINO",
description="open-set object detector",
@@ -215,6 +215,6 @@
"tests",
)
),
- ext_modules=get_extensions(),
- cmdclass={"build_ext": torch.utils.cpp_extension.BuildExtension},
+ # ext_modules=get_extensions(),
+ # cmdclass={"build_ext": torch.utils.cpp_extension.BuildExtension},
)
