Use GPU based on config for voigt.py

smokestacklightnin · Sep 20, 2023 · c5244b3 · c5244b3
1 parent 6627f81
commit c5244b3
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/stardis/radiation_field/opacities/opacities_solvers/voigt.py b/stardis/radiation_field/opacities/opacities_solvers/voigt.py
@@ -2,8 +2,18 @@
 import numba
 from numba import cuda
 import cmath
+from pathlib import Path
+from importlib.machinery import PathFinder
 
-GPUs_available = cuda.is_available()
+from hydra import compose, initialize
+
+with initialize(
+    version_base=None,
+    config_path=pathlib.Path(m.PathFinder.find_spec("stardis").origin).parent.parent,
+    job_name="stardis",
+):
+    cfg = compose(config_name="config")
+GPUs_available = cfg.USE_GPU
 
 if GPUs_available:
     import cupy as cp