WIP

Author: govindchari

CMakeLists.txt

@@ -29,6 +29,8 @@ FetchContent_Declare(
 list(POP_BACK CMAKE_MESSAGE_INDENT)
 FetchContent_MakeAvailable(qoco)
 
+configure_file(${CMAKE_CURRENT_SOURCE_DIR}/src/bindings.cpp.in
+               ${CMAKE_CURRENT_SOURCE_DIR}/src/bindings.cpp)
 pybind11_add_module(qoco_ext src/bindings.cpp)
 target_include_directories(qoco_ext INTERFACE ${qoco_SOURCE_DIR}/include)
 

backends/cuda/pyproject.toml

@@ -0,0 +1,24 @@
+[build-system]
+requires = ["scikit-build-core", "pybind11"]
+build-backend = "scikit_build_core.build"
+
+[project]
+name = "qoco"
+version = "0.2.0"
+description = "QOCO: Quadratic Objective Conic Optimizer"
+readme = "README.md"
+requires-python = ">=3.8"
+authors = [{ name = "Govind M. Chari", email = "govindchari1@gmail.com" }]
+dependencies = ["jinja2", "numpy>=1.7", "qdldl", "scipy>=0.13.2", "setuptools"]
+
+[tool.scikit-build]
+install.components = ["python"]
+wheel.install-dir = "qoco"
+
+[tool.scikit-build.cmake.define]
+QOCO_ALGEBRA_BACKEND = "cuda"
+QOCO_EXT_MODULE_NAME = "qoco_cuda"
+
+[project.urls]
+Homepage = "https://github.com/qoco-org/qoco"
+Issues = "https://github.com/qoco-org/qoco/issues"

pyproject.toml

@@ -16,7 +16,8 @@ install.components = ["python"]
 wheel.install-dir = "qoco"
 
 [tool.scikit-build.cmake.define]
-QOCO_ALGEBRA_BACKEND = "cuda"
+QOCO_ALGEBRA_BACKEND = "builtin"
+QOCO_EXT_MODULE_NAME = "qoco_ext"
 
 [project.urls]
 Homepage = "https://github.com/qoco-org/qoco"

src/bindings.cpp src/bindings.cpp.insrc/bindings.cpp renamed to src/bindings.cpp.in

@@ -241,7 +241,7 @@ QOCOInt PyQOCOSolver::update_settings(const QOCOSettings &new_settings)
     return qoco_update_settings(this->_solver, &new_settings);
 }
 
-PYBIND11_MODULE(qoco_ext, m)
+PYBIND11_MODULE(@QOCO_EXT_MODULE_NAME@, m)
 {
     // Enums.
     py::enum_<qoco_solve_status>(m, "qoco_solve_status", py::module_local())

src/qoco/interface.py

@@ -5,7 +5,32 @@
 import numpy as np
 from scipy import sparse
 from types import SimpleNamespace
-import time
+
+ALGEBRAS = (
+    "cuda",
+    "builtin",
+)
+
+ALGEBRA_MODULES = {
+    "cuda": "qoco_cuda",
+    "builtin": "qoco.qoco_ext",
+}
+
+
+def algebra_available(algebra):
+    assert algebra in ALGEBRAS, f"Unknown algebra {algebra}"
+    module = ALGEBRA_MODULES[algebra]
+
+    try:
+        importlib.import_module(module)
+    except ImportError:
+        return False
+    else:
+        return True
+
+
+def algebras_available():
+    return [algebra for algebra in ALGEBRAS if algebra_available(algebra)]
 
 
 class QOCO:
@@ -41,7 +66,10 @@ def __init__(self, *args, **kwargs):
             "QOCO_MAX_ITER",
         ]
 
-        self.ext = importlib.import_module("qoco.qoco_ext")
+        self.algebra = kwargs.pop("algebra") if "algebra" in kwargs else "builtin"
+        if not algebra_available(self.algebra):
+            raise RuntimeError(f"Algebra {self.algebra} not available")
+        self.ext = importlib.import_module(ALGEBRA_MODULES[self.algebra])
         self._solver = None
 
     def update_settings(self, **kwargs):