fix diso, spconv and mesh_painter installation

_Pre_Builds/_Build_Scripts/auto_build_all.py

@@ -52,11 +52,6 @@ def setup_build_env():
     
     install_remote_packages(build_config.build_base_packages)
     install_platform_packages()
-    
-    # Install spconv with correct CUDA version
-    from build_utils import install_spconv
-    if not install_spconv():
-        raise RuntimeError("Failed to install spconv")
 
 def clone_or_update_read_dependency(dependency_url, dependency_dir):
     if os.path.exists(dependency_dir):

_Pre_Builds/_Build_Scripts/build_config.yaml

@@ -13,7 +13,10 @@ libs_dir_name: "_Libs"
 # Folder used to store built wheels
 wheels_dir_name: "_Build_Wheels"
 
-build_base_packages: ["torch", "torchvision", "xformers"]
+build_base_packages: ["torch", "torchvision", "torchaudio", "xformers"]
+
+# Packages requiring special install flags (like --no-build-isolation)
+isolated_packages: ["diso"]
 
 # Platform-specific packages
 platform_packages:
@@ -40,10 +43,18 @@ remote_packages:
     version: "0.22.0"
     url: "https://download.pytorch.org/whl/${cuda_version}"
 
+  torchaudio:
+    version: "2.7.0"
+    url: "https://download.pytorch.org/whl/${cuda_version}"
+
   xformers:
     version: "0.0.30"
     url: "https://download.pytorch.org/whl/${cuda_version}"
 
   torch-scatter:
    url_option: "--find-links"
-   url: "https://data.pyg.org/whl/torch-${remote_packages.torch.version}+${cuda_version}.html"
+   url: "https://data.pyg.org/whl/torch-${remote_packages.torch.version}+${cuda_version}.html"
+  
+  diso:
+    url: "git+https://github.com/SarahWeiii/diso.git#egg=diso"
+    install_flags: ["--no-build-isolation"]

_Pre_Builds/_Build_Scripts/build_utils.py

@@ -250,68 +250,31 @@ def install_platform_packages():
             print(f"Installing platform package {package}...")
             subprocess.run([PYTHON_PATH, "-s", "-m", "pip", "install", package])
 
-def install_spconv():
-    """Simple spconv installation with correct CUDA version"""
-    # Check if spconv is already installed and working
-    try:
-        import spconv.core_cc
-        print("spconv is already installed and working")
-        return True
-    except ImportError:
-        pass
-    
-    # Get CUDA version mapping
-    if not hasattr(build_config, 'spconv_cuda_mapping'):
-        print("No spconv_cuda_mapping in config, skipping spconv installation")
-        return True
-    
-    # Fix CUDA version parsing: cu128 -> 12.8, cu121 -> 12.1, cu118 -> 11.8
-    if len(CUDA_VERSION) >= 5:  # e.g., cu128, cu121
-        major = CUDA_VERSION[2:-1]  # cu128 -> 12, cu121 -> 12, cu118 -> 11  
-        minor = CUDA_VERSION[-1]    # cu128 -> 8, cu121 -> 1, cu118 -> 8
-        cuda_version_for_lookup = f"{major}.{minor}"
-    else:
-        print(f"Invalid CUDA_VERSION format: {CUDA_VERSION}")
-        return False
-    
-    print(f"Detected CUDA version: {cuda_version_for_lookup}")
-    
-    cuda_suffix = build_config.spconv_cuda_mapping.get(cuda_version_for_lookup)
-    
-    if not cuda_suffix:
-        print(f"No spconv mapping for CUDA {cuda_version_for_lookup}, skipping")
-        return True
-    
-    spconv_package = f"spconv-{cuda_suffix}"
-    print(f"Installing {spconv_package}...")
-    
-    result = subprocess.run(
-        [PYTHON_PATH, "-m", "pip", "install", spconv_package],
-        text=True, capture_output=True
-    )
-    
-    if result.returncode != 0:
-        print(f"Failed to install {spconv_package}: {result.stderr}")
-        return False
-    
-    # Verify installation
-    try:
-        import spconv.core_cc
-        print(f"Successfully installed and verified {spconv_package}")
-        return True
-    except ImportError as e:
-        print(f"spconv installed but verification failed: {e}")
-        return False
-
-def check_spconv_after_wheels():
-    """Check if spconv works after wheels installation"""
-    try:
-        import spconv.core_cc
-        print("spconv is working after wheels installation")
-        return True
-    except ImportError:
-        print("spconv not working, will install from mapping")
-        return False
+def install_isolated_packages(package_names):
+    """Install packages with special flags like --no-build-isolation"""
+    for package_name in package_names:
+        if package_name in build_config.remote_packages:
+            package_attr = build_config.remote_packages[package_name]
+            
+            # Check if package is already installed
+            if is_package_installed(package_name):
+                print(f"Package {package_name} is already installed, skipping...")
+                continue
+            
+            print(f"Installing isolated package {package_name}...")
+            
+            if hasattr(package_attr, "url"):
+                # Build command with install flags
+                cmd = [PYTHON_PATH, "-s", "-m", "pip", "install"]
+                if hasattr(package_attr, "install_flags"):
+                    cmd.extend(package_attr.install_flags)
+                cmd.append(package_attr.url)
+                
+                subprocess.run(cmd)
+            else:
+                print(f"No URL found for isolated package {package_name}")
+        else:
+            print(f"Isolated package {package_name} not found in config")
 
 def wheels_dir_exists_and_not_empty(builds_dir):
     if not os.path.exists(builds_dir):