fix: build setup and JIT loading for Blackwell architecture (#177)

* fix: add --no-build-isolation to pip install because of blackwell architecture
* fix: refactor JIT module loading to explicitly return modules (needed for blackwell architecture)
* fix: missing --no-build-isolation flag

---------

Co-authored-by: Qi Wu <wilson.over.cloud@gmail.com>

Author: ahamboeck

install_env.sh

@@ -173,7 +173,7 @@ conda install -c conda-forge mesa-libgl-devel-cos7-x86_64 -y
 
 # Initialize git submodules and install Python requirements
 git submodule update --init --recursive
-pip install -r requirements.txt
-pip install -e .
+pip install --no-build-isolation -r requirements.txt
+pip install --no-build-isolation -e .
 
 echo "Setup completed successfully!"

threedgrt_tracer/setup_3dgrt.py

@@ -84,4 +84,4 @@ def to_cpp_bool(value):
         extra_cflags=cflags,
         extra_cuda_cflags=cuda_flags,
         extra_include_paths=include_paths,
-    )
+    )

threedgrut/gui/setup_gui.py

@@ -41,4 +41,4 @@ def setup_gui():
         extra_cflags=cflags,
         extra_cuda_cflags=cuda_cflags,
         extra_include_paths=include_paths,
-    )
+    )

threedgrut/strategy/mcmc.py

@@ -39,11 +39,10 @@ def load_mcmc_plugin():
     if _mcmc_plugin is None:
         try:
             from . import lib_mcmc_cc as gaussian_mcmc
+            _mcmc_plugin = gaussian_mcmc  # type: ignore
         except ImportError:
             from threedgrut.strategy.src.setup_mcmc import setup_mcmc
-
-            gaussian_mcmc = setup_mcmc()
-        _mcmc_plugin = gaussian_mcmc
+            _mcmc_plugin = setup_mcmc()
 
 
 class MCMCStrategy(BaseStrategy):

threedgrut/strategy/src/setup_mcmc.py

@@ -56,4 +56,4 @@ def setup_mcmc():
         extra_cuda_cflags=cuda_cflags,
         extra_include_paths=include_paths,
         build_directory=build_dir,
-    )
+    )

threedgrut_playground/setup_playground.py

@@ -87,4 +87,4 @@ def to_cpp_bool(value):
         name="libplayground_cc",
         sources=source_paths,
         extra_include_paths=include_paths,
-    )
+    )

threedgut_tracer/setup_3dgut.py

@@ -142,4 +142,4 @@ def to_cpp_bool(value):
         extra_cuda_cflags=cuda_cflags,
         extra_include_paths=include_paths,
         build_directory=build_dir,
-    )
+    )