Updating build system as scikit-build.

Author: tatsy

.github/workflows/build.yml

@@ -2,8 +2,6 @@ name: Build (CPU)
 on:
   push:
     branches: [master, dev]
-  # release:
-  #   types: [published]
 
 jobs:
   build:
@@ -22,26 +20,14 @@ jobs:
     steps:
       - name: Check out repository code
         uses: actions/checkout@v4
-      - name: Install Poetry
-        run: |
-          python -m pip install --upgrade pip
-          python -m pip install poetry
       - name: Setup Python
         uses: actions/setup-python@v4
         with:
           python-version: ${{ matrix.version }}
-          cache: "poetry"
-      - name: Install dependencies
+      - name: Install requirements
         run: |
-          poetry install
+          python -m pip install --upgrade pip
+          python -m pip install numpy "pybind11[global]"
+          python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
       - name: Build
-        run: |
-          poetry install
-      # - name: Check
-      #   run: poetry run python -c "import cshogi"
-      # - name: Publish
-      #   if: github.event_name == 'release' && github.event.action == 'published'
-      #   env:
-      #     TWINE_USERNAME: ${{ secrets.PYPI_USERNAME }}
-      #     TWINE_PASSWORD: ${{ secrets.PYPI_PASSWORD }}
-      #   run: twine upload dist/*
+        run: pip install .

.gitignore

@@ -1,9 +1,9 @@
-build/*
-dist/*
-.vscode/*
-.mypy_cache/*
-
-__pycache__
-*.pyd
-*.so
-*.egg-info
+build
+_skbuild
+dist
+.mypy_cache
+
+__pycache__
+*.pyd
+*.so
+*.egg-info

.vscode/settings.json

@@ -0,0 +1,12 @@
+{
+    "flake8.path": "pflake8",
+    "[python]": {
+        "editor.formatOnSave": true,
+        "editor.formatOnPaste": false,
+        "editor.formatOnType": false,
+        "editor.defaultFormatter": "eeyore.yapf",
+        "editor.codeActionsOnSave": {
+            "source.organizeImports": "always",
+        }
+    }
+}

CMakeLists.txt

@@ -0,0 +1,57 @@
+cmake_minimum_required(VERSION 3.15...3.27)
+project(${SKBUILD_PROJECT_NAME} VERSION ${SKBUILD_PROJECT_VERSION} LANGUAGES CXX)
+
+# General settings
+set(BUILD_TARGET torchmcubes_module)
+
+# CUDA settings
+include(CheckLanguage)
+check_language(CUDA)
+
+if(CMAKE_CUDA_COMPILER)
+  set(CMAKE_CUDA_ARCHITECTURES "native")
+  enable_language(CUDA)
+
+  add_definitions(-DWITH_CUDA)
+  find_package(CUDAToolkit REQUIRED)
+  set(CMAKE_CUDA_STANDARD 17)
+  set(CMAKE_CUDA_STANDARD_REQUIRED ON)
+
+  message(STATUS "INSTALLING EXTENSIONS WITH CUDA!")
+  string(REGEX REPLACE ".[0-9][0-9]|\\." "" CUDA_V ${CMAKE_CUDA_COMPILER_VERSION})
+  message(STATUS "CMAKE_CUDA_COMPILER = ${CMAKE_CUDA_COMPILER}")
+  message(STATUS "CMAKE_CUDA_COMPILER_ID = ${CMAKE_CUDA_COMPILER_ID}")
+  message(STATUS "CMAKE_CUDA_COMPILER_VERSION = ${CUDA_V}")
+else()
+  message(STATUS "NO CUDA INSTALLATION FOUND, INSTALLING CPU VERSION ONLY!")
+  # execute_process(COMMAND ${Python_EXECUTABLE} -m pip uninstall -y torch)
+  # execute_process(COMMAND ${Python_EXECUTABLE} -m pip install -v torch --extra-index-url https://download.pytorch.org/whl/cpu)
+endif()
+
+set(CMAKE_CXX_STANDARD 17)
+set(CMAKE_CXX_STANDARD_REQUIRED ON)
+set(CMAKE_POSITION_INDEPENDENT_CODE ON)
+
+if(MSVC)
+  add_definitions(-DNOMINMAX)
+endif()
+
+# Find packages
+find_package(Python REQUIRED COMPONENTS Interpreter Development)
+find_package(pybind11 CONFIG REQUIRED)
+
+# OpenMP
+find_package(OpenMP)
+if (OPENMP_FOUND)
+  set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
+  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
+endif()
+
+# PyTorch settings
+find_package(Torch REQUIRED)
+find_library(TORCH_PYTHON_LIBRARY torch_python PATH "${TORCH_INSTALL_PREFIX}/lib")
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")
+
+add_subdirectory(cxx)
+
+install(TARGETS ${BUILD_TARGET} LIBRARY DESTINATION .)