cunibs 0.1.0__tar.gz

cunibs-0.1.0/.github/workflows/wheels.yml

@@ -0,0 +1,84 @@
+name: Build wheels
+
+on:
+  push:
+    tags:
+      - "v*"
+
+permissions:
+  contents: read
+
+jobs:
+  build-wheels:
+    name: Build ${{ matrix.python }} wheel
+    runs-on: ubuntu-24.04
+    strategy:
+      matrix:
+        python:
+          - cp312
+          - cp313
+          - cp314
+          - cp314t
+
+    steps:
+      - uses: actions/checkout@v7.0.0
+        with:
+          persist-credentials: false
+          submodules: recursive
+
+      - uses: pypa/cibuildwheel@v4.1.0
+        env:
+          CIBW_ARCHS_LINUX: x86_64
+          CIBW_BUILD: ${{ matrix.python }}-manylinux_x86_64
+          CIBW_ENVIRONMENT_LINUX: >-
+            CMAKE_ARGS="-DCMAKE_CUDA_ARCHITECTURES=75;80;86;89;90;100;120"
+          CIBW_REPAIR_WHEEL_COMMAND_LINUX: >-
+            auditwheel repair -w {dest_dir}
+            --exclude 'libcudart.so.*'
+            --exclude 'libcublas.so.*'
+            --exclude 'libcusparse.so.*'
+            --exclude 'libcusolver.so.*'
+            {wheel}
+
+      - uses: actions/upload-artifact@v7.0.1
+        with:
+          name: wheels-${{ matrix.python }}-manylinux-x86_64
+          path: wheelhouse/*.whl
+          if-no-files-found: error
+
+  build-sdist:
+    name: Build sdist
+    runs-on: ubuntu-24.04
+    steps:
+      - uses: actions/checkout@v7.0.0
+        with:
+          persist-credentials: false
+          submodules: recursive
+
+      - uses: astral-sh/setup-uv@v8.2.0
+      - run: uv build --sdist
+
+      - uses: actions/upload-artifact@v7.0.1
+        with:
+          name: wheels-sdist
+          path: dist/*.tar.gz
+          if-no-files-found: error
+
+  publish:
+    name: Publish to PyPI
+    needs: [build-wheels, build-sdist]
+    runs-on: ubuntu-24.04
+    if: startsWith(github.ref, 'refs/tags/')
+    environment:
+      name: pypi
+      url: https://pypi.org/p/cunibs
+    permissions:
+      id-token: write
+    steps:
+      - uses: actions/download-artifact@v8.0.1
+        with:
+          pattern: wheels-*
+          path: dist
+          merge-multiple: true
+
+      - uses: pypa/gh-action-pypi-publish@release/v1

cunibs-0.1.0/.gitignore

@@ -0,0 +1,15 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv
+.python-version
+
+# Tooling
+.pytest_cache
+.ruff_cache

cunibs-0.1.0/.gitmodules

@@ -0,0 +1,3 @@
+[submodule "external/AMGX"]
+	path = external/AMGX
+	url = https://github.com/srimanachanta/AMGX

cunibs-0.1.0/CMakeLists.txt

@@ -0,0 +1,99 @@
+cmake_minimum_required(VERSION 3.21)
+
+find_package(Python 3.12 COMPONENTS Interpreter Development.Module REQUIRED)
+
+execute_process(
+  COMMAND "${Python_EXECUTABLE}" -c "import nvidia.cu13 as m, os; print(os.path.join(m.__path__[0], 'bin', 'nvcc'))"
+  OUTPUT_STRIP_TRAILING_WHITESPACE OUTPUT_VARIABLE CMAKE_CUDA_COMPILER
+)
+
+if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
+  set(CMAKE_CUDA_ARCHITECTURES native)
+endif()
+
+project(${SKBUILD_PROJECT_NAME} LANGUAGES CXX CUDA)
+
+get_filename_component(_CUDA_BIN_DIR "${CMAKE_CUDA_COMPILER}" DIRECTORY)
+set(_CUDA_LIB_DIR "${_CUDA_BIN_DIR}/../lib")
+set(_CUDA_SHIM_LIBS cudart cublas cusparse cusolver)
+foreach(_lib ${_CUDA_SHIM_LIBS})
+  if(EXISTS "${_CUDA_LIB_DIR}" AND NOT EXISTS "${_CUDA_LIB_DIR}/lib${_lib}.so")
+    file(GLOB _versioned "${_CUDA_LIB_DIR}/lib${_lib}.so.[0-9]*")
+    list(LENGTH _versioned _count)
+    if(_count GREATER 0)
+      list(GET _versioned 0 _first)
+      file(CREATE_LINK "${_first}" "${_CUDA_LIB_DIR}/lib${_lib}.so" SYMBOLIC)
+    endif()
+  endif()
+endforeach()
+
+find_package(CUDAToolkit 13.2 REQUIRED)
+
+set(CMAKE_CXX_STANDARD 20)
+set(CMAKE_CXX_STANDARD_REQUIRED ON)
+set(CMAKE_CUDA_STANDARD 20)
+set(CMAKE_CUDA_STANDARD_REQUIRED ON)
+set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
+
+if(NOT CMAKE_BUILD_TYPE AND NOT CMAKE_CONFIGURATION_TYPES)
+  set(CMAKE_BUILD_TYPE "Release" CACHE STRING "Default build type" FORCE)
+endif()
+
+set(AMGX_BUILD_TESTS OFF CACHE BOOL "" FORCE)
+set(AMGX_BUILD_EXAMPLES OFF CACHE BOOL "" FORCE)
+set(AMGX_BUILD_STATIC OFF CACHE BOOL "" FORCE)
+set(AMGX_BUILD_SHARED ON  CACHE BOOL "" FORCE)
+set(AMGX_INSTALL OFF CACHE BOOL "" FORCE)
+set(CMAKE_NO_MPI ON CACHE BOOL "" FORCE)
+set(AMGX_BUILD_DEVICE_MODE AMGX_mode_dDDI CACHE STRING "" FORCE)
+set(AMGX_BUILD_HOST_MODE AMGX_mode_hDDI CACHE STRING "" FORCE)
+set(AMGX_ENABLE_COMPLEX OFF CACHE BOOL "" FORCE)
+set(AMGX_ENABLE_MIXED_PRECISION OFF CACHE BOOL "" FORCE)
+set(AMGX_MINIMAL_SOLVERS ON CACHE BOOL "" FORCE)
+
+add_subdirectory(external/AMGX) 
+
+target_compile_options(amgx_libs PRIVATE
+  $<$<COMPILE_LANGUAGE:CUDA>:-Xcompiler=-Wno-deprecated-declarations>
+  $<$<COMPILE_LANGUAGE:CXX>:-Wno-deprecated-declarations>)
+
+execute_process(
+  COMMAND "${Python_EXECUTABLE}" -m nanobind --cmake_dir
+  OUTPUT_STRIP_TRAILING_WHITESPACE OUTPUT_VARIABLE nanobind_ROOT)
+find_package(nanobind CONFIG REQUIRED)
+
+nanobind_add_module(_solver_ext LTO FREE_THREADED
+  src/cunibs/solver/bindings.cpp
+  src/cunibs/solver/solver.cpp
+  src/cunibs/solver/dadt.cu
+  src/cunibs/solver/rhs.cu
+  src/cunibs/solver/reconstruct.cu
+)
+target_link_libraries(_solver_ext PRIVATE CUDA::cudart AMGX::amgxsh)
+
+set_target_properties(_solver_ext amgxsh PROPERTIES
+  INSTALL_RPATH "$ORIGIN;$ORIGIN/../../nvidia/cu13/lib"
+)
+
+get_target_property(_amgxsh_soversion amgxsh SOVERSION)
+set_target_properties(amgxsh PROPERTIES VERSION ${_amgxsh_soversion})
+
+install(TARGETS _solver_ext amgxsh
+  LIBRARY DESTINATION cunibs/solver
+  NAMELINK_SKIP
+)
+
+nanobind_add_stub(
+  _solver_ext_stub
+  MODULE _solver_ext
+  OUTPUT solver/_solver_ext.pyi
+  PYTHON_PATH $<TARGET_FILE_DIR:_solver_ext>
+  DEPENDS _solver_ext
+  MARKER_FILE solver/py.typed
+)
+
+install(FILES
+  ${CMAKE_CURRENT_BINARY_DIR}/solver/_solver_ext.pyi
+  ${CMAKE_CURRENT_BINARY_DIR}/solver/py.typed
+  DESTINATION cunibs/solver
+)