From 79e1abb2ff825057beefb43952d19c10305e0853 Mon Sep 17 00:00:00 2001
From: AndreaRigoni <andrea.rigoni@igi.cnr.it>
Date: Thu, 5 Mar 2026 15:17:30 +0000
Subject: [PATCH] add USE_CUDA env in python_build

---
 README.md       | 7 +++++++
 build_python.py | 9 ++++++++-
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 7ebab60..b765451 100644
--- a/README.md
+++ b/README.md
@@ -61,3 +61,10 @@ cmake --preset conan-release
 ```bash
 cmake --build build -j10
 ```
+
+### Make python package
+
+```bash
+micromamba run -n mutom env USE_CUDA=ON poetry install
+```
+
diff --git a/build_python.py b/build_python.py
index 1a8554f..5ab1b4f 100644
--- a/build_python.py
+++ b/build_python.py
@@ -26,12 +26,19 @@ def build(setup_kwargs):
     print(f"Project root: {project_root}")
     print(f"Target binary dir: {package_dir}")
 
+    # Determine if CUDA should be enabled
+    use_cuda = os.environ.get("USE_CUDA", "OFF").upper()
+    if use_cuda in ["ON", "1", "TRUE", "YES"]:
+        use_cuda = "ON"
+    else:
+        use_cuda = "OFF"
+    
     # CMake configuration
     cmake_args = [
         f"-DCMAKE_LIBRARY_OUTPUT_DIRECTORY={package_dir}",
         f"-DPYTHON_EXECUTABLE={sys.executable}",
         "-DCMAKE_BUILD_TYPE=Release",
-        "-DUSE_CUDA=OFF", 
+        f"-DUSE_CUDA={use_cuda}", 
         "-G", "Unix Makefiles",
     ]