Merge up-to-date changes into main

.gitignore

@@ -127,3 +127,12 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+
+# Pycharm
+.idea/*
+
+# Development directory
+__dev
+
+# Binary files
+libs/ccc/sklearn/metrics

.pre-commit-config.yaml

@@ -0,0 +1,20 @@
+# See https://pre-commit.com for more information
+# See https://pre-commit.com/hooks.html for more hooks
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v5.0.0
+    hooks:
+      - id: trailing-whitespace
+      - id: end-of-file-fixer
+      - id: check-yaml
+      - id: check-toml
+      - id: check-added-large-files
+      - id: mixed-line-ending
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.8.5  # Ruff version.
+    hooks:
+      - id: ruff  # Run the linter.
+        types_or: [python, pyi]
+        args: [--fix]
+      - id: ruff-format  # Run the formatter.
+        types_or: [python, pyi]

.readthedocs.yaml

@@ -0,0 +1,35 @@
+# Read the Docs configuration file for Sphinx projects
+# See https://docs.readthedocs.io/en/stable/config-file/v2.html for details
+
+# Required
+version: 2
+
+# Set the OS, Python version and other tools you might need
+build:
+  os: ubuntu-22.04
+  tools:
+    python: "3.12"
+    # You can also specify other tool versions:
+    # nodejs: "20"
+    # rust: "1.70"
+    # golang: "1.20"
+
+# Build documentation in the "docs/" directory with Sphinx
+sphinx:
+  configuration: docs/source/conf.py
+  # You can configure Sphinx to use a different builder, for instance use the dirhtml builder for simpler URLs
+  # builder: "dirhtml"
+  # Fail on all warnings to avoid broken references
+  # fail_on_warning: true
+
+# Optionally build your docs in additional formats such as PDF and ePub
+# formats:
+#   - pdf
+#   - epub
+
+# Optional but recommended, declare the Python requirements required
+# to build your documentation
+# See https://docs.readthedocs.io/en/stable/guides/reproducible-builds.html
+# python:
+#   install:
+#     - requirements: docs/requirements.txt

CMakeLists.txt

@@ -0,0 +1,100 @@
+cmake_minimum_required(VERSION 3.15...3.26)
+project(${SKBUILD_PROJECT_NAME} LANGUAGES CUDA CXX)
+
+# Add this near the top of your file, after project()
+# Define the include directories for the whole project
+set(PROJECT_INCLUDE_DIR ${CMAKE_SOURCE_DIR}/libs)
+# Set extention name and source directory
+set(CUDA_EXT_MODULE_NAME ccc_cuda_ext)
+set(CUDA_EXT_DIR ${CMAKE_CURRENT_SOURCE_DIR}/libs/${CUDA_EXT_MODULE_NAME})
+
+# Set C++ standard
+set(CMAKE_CXX_STANDARD 20)
+set(CMAKE_CXX_STANDARD_REQUIRED ON)
+
+# Set Python Standard
+# Get Python version dynamically
+execute_process(
+    COMMAND "${Python_EXECUTABLE}" -c "import sys; print(f'{sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}')"
+    OUTPUT_VARIABLE PYTHON_VERSION_FULL
+    OUTPUT_STRIP_TRAILING_WHITESPACE
+)
+# Set the paths using the detected version
+set(Python_EXECUTABLE $ENV{CONDA_PREFIX}/bin/python)
+set(PYTHON_INCLUDE_DIR $ENV{CONDA_PREFIX}/include/python${PYTHON_VERSION_FULL})
+set(PYTHON_LIBRARY $ENV{CONDA_PREFIX}/lib/libpython${PYTHON_VERSION_FULL}.so)
+
+
+# Set CUDA architecture and Pybind11
+find_package(Python REQUIRED Development)
+set(PYBIND11_NEWPYTHON ON)
+find_package(pybind11 CONFIG REQUIRED)
+
+# Download and configure Google Test
+include(FetchContent)
+FetchContent_Declare(
+    googletest
+    GIT_REPOSITORY https://github.com/google/googletest.git
+    GIT_TAG v1.15.2  # Adjust version as needed
+)
+FetchContent_MakeAvailable(googletest)
+
+# Setup Gtest
+enable_testing()
+# Function to automatically add tests from a directory
+function(add_tests_from_directory TEST_DIR)
+    # Find all test files in the directory
+    file(GLOB_RECURSE TEST_FILES 
+        "${TEST_DIR}/*_test.cpp"    # Files ending with _test.cpp
+        "${TEST_DIR}/*_tests.cpp"   # Files ending with _tests.cpp
+        "${TEST_DIR}/test_*.cpp"    # Files starting with test_
+    )
+
+    # Loop through each test file
+    foreach(TEST_FILE ${TEST_FILES})
+        # Get the filename without extension
+        get_filename_component(TEST_NAME ${TEST_FILE} NAME_WE)
+
+        # Create an executable for this test
+        add_executable(${TEST_NAME} ${TEST_FILE} ${CUDA_EXT_DIR}/metrics.cu)
+
+        # target_include_directories(${TEST_NAME} PRIVATE
+        #     ${PROJECT_INCLUDE_DIR}  # Add this line
+        #     ${Python_INCLUDE_DIRS}
+        # )
+
+        # Link against gtest and your project libraries
+        target_link_libraries(${TEST_NAME} PRIVATE
+            GTest::gtest_main
+            GTest::gtest
+            pybind11::headers
+            pybind11::embed
+            Python::Python
+            # Add your other project libraries here
+            # project_lib
+        )
+
+        # Add the test to CTest
+        add_test(NAME ${TEST_NAME} COMMAND ${TEST_NAME})
+
+        # Set test properties (optional)
+        # Set test properties (optional)
+        set_tests_properties(${TEST_NAME} PROPERTIES
+            TIMEOUT 10  # Timeout in seconds
+            WORKING_DIRECTORY "${CMAKE_RUNTIME_OUTPUT_DIRECTORY}"
+        )
+    endforeach()
+endfunction()
+
+# Specify your test directory and call the function
+add_tests_from_directory(${CMAKE_CURRENT_SOURCE_DIR}/tests)
+
+# Optional: Set output directories
+set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/bin)
+set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/lib)
+
+
+# Set up binding then do compilation and installation
+pybind11_add_module(${CUDA_EXT_MODULE_NAME} ${CUDA_EXT_DIR}/binder.cu ${CUDA_EXT_DIR}/metrics.cu)
+
+install(TARGETS ${CUDA_EXT_MODULE_NAME} LIBRARY DESTINATION .)