Update src/utils/dependencies/XPose/models/UniPose/ops/setup.py

src/utils/dependencies/XPose/models/UniPose/ops/setup.py

@@ -27,36 +27,25 @@ def get_extensions():
 
     main_file = glob.glob(os.path.join(extensions_dir, "*.cpp"))
     source_cpu = glob.glob(os.path.join(extensions_dir, "cpu", "*.cpp"))
-    source_cuda = glob.glob(os.path.join(extensions_dir, "cuda", "*.cu"))
+    # source_cuda = glob.glob(os.path.join(extensions_dir, "cuda", "*.cu")) # Don't include CUDA sources
 
+    # Force CPU Build Path
+    print("Forcing CPU build path for MultiScaleDeformableAttention.")
     sources = main_file + source_cpu
-    extension = CppExtension
+    extension = CppExtension # Force CppExtension
     extra_compile_args = {"cxx": []}
-    define_macros = []
+    define_macros = [] # No CUDA macros
 
-    # import ipdb; ipdb.set_trace()
-
-    if torch.cuda.is_available() and CUDA_HOME is not None:
-        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:
-        # Allow build to continue even if CUDA not detected initially
-        print("Warning: CUDA not found during build setup for MultiScaleDeformableAttention. Attempting build anyway, may fall back to CPU.")
-        pass # Avoid raising error
+    # Remove the CUDA check block entirely
+    # if torch.cuda.is_available() and CUDA_HOME is not None:
+    #    ...
+    # else:
+    #    ...
 
     sources = [os.path.join(extensions_dir, s) for s in sources]
     include_dirs = [extensions_dir]
-    # Ensure extension type is set correctly based on initial check
-    extension_type = CUDAExtension if extension == CUDAExtension else CppExtension
     ext_modules = [
-        extension_type( # Use the determined extension type here
+        extension( # Use the forced CppExtension type
             "MultiScaleDeformableAttention",
             sources,
             include_dirs=include_dirs,