From 0eedbf4dc459192e8ae23b21db0b5d2525f3f5df Mon Sep 17 00:00:00 2001 From: youben11 Date: Wed, 28 Sep 2022 15:07:14 +0100 Subject: [PATCH] build(python): add opt to build with CUDA support env variables can be set to build python wheels with CUDA support --- compiler/setup.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/compiler/setup.py b/compiler/setup.py index 2618f1a09..bcca6c28f 100644 --- a/compiler/setup.py +++ b/compiler/setup.py @@ -42,6 +42,12 @@ class MakeBuild(build_ext): py_exec = os.environ.get("CONCRETE_COMPILER_Python3_EXECUTABLE") if py_exec: cmd.append(f"Python3_EXECUTABLE={py_exec}") + cuda_support = os.environ.get("CONCRETE_COMPILER_CUDA_SUPPORT") + if cuda_support: + cmd.append(f"CUDA_SUPPORT={cuda_support}") + concrete_core_path = os.environ.get("CONCRETE_COMPILER_CONCRETE_CORE_PATH") + if concrete_core_path: + cmd.append(f"CONCRETE_CORE_PATH={concrete_core_path}") cmd.append(f"BUILD_DIR={build_dir()}") cmd.append("python-bindings") subprocess.check_call(cmd)