keopscore config / cpp compilation

.github/workflows/black.yml

@@ -1,12 +1,19 @@
-name: Lint
+name: Format Check
 
-on: [push, pull_request]
+on:
+  push:
+  pull_request:
 
 jobs:
-  lint:
+  format:
+    name: Format (Black)
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v2
-      - uses: actions/setup-python@v2
-      - uses: psf/black@stable
+      - name: Checkout
+        uses: actions/checkout@v4
+
+      - name: Black format check
+        uses: psf/black@stable
+        with:
+          options: "--check ."
 

.github/workflows/linux_cpu_test.yml

@@ -0,0 +1,21 @@
+name: Linux CPU Test
+
+on:
+  push:
+  pull_request:
+
+jobs:
+  tests-linux:
+    name: Tests (Linux CPU)
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.x"
+
+      - name: Run tests
+        run: bash ./pytest.sh

.github/workflows/linux_cuda_test.yml

@@ -0,0 +1,16 @@
+name: Linux CUDA Test
+
+on:
+  push:
+  pull_request:
+
+jobs:
+  tests-linux-cuda:
+    name: Tests (Linux CUDA)
+    runs-on: self-hosted
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+
+      - name: Run tests
+        run: bash ./pytest.sh

.github/workflows/macOs_test.yml

@@ -0,0 +1,46 @@
+name: macOS Test
+
+on:
+  push:
+  pull_request:
+
+jobs:
+  tests-macos-latest:
+    name: Tests (macOS latest)
+    runs-on: macos-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.x"
+
+      - name: Install dependencies
+        run: brew install libomp
+
+      - name: Run tests
+        run: bash ./pytest.sh
+
+  tests-macos-intel:
+    name: Tests (macOS Intel)
+    runs-on: macos-26-intel
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+
+      - name: Pin NumPy for Intel CI
+        run: |
+          echo "numpy<2" > /tmp/pip-constraints.txt
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.10"
+
+      - name: Install dependencies
+        run: brew install libomp
+
+      - name: Run tests
+        run: bash ./pytest.sh --pip-constraint=/tmp/pip-constraints.txt -v 2

.pre-commit-config.yaml

@@ -0,0 +1,14 @@
+repos:
+  - repo: https://github.com/psf/black
+    rev: 24.10.0
+    hooks:
+      - id: black
+
+  - repo: local
+    hooks:
+      - id: pytest
+        name: pytest
+        entry: ./pytest.sh
+        language: system
+        pass_filenames: false
+        always_run: true

benchmarks/PyTorch_GPU.py

@@ -12,7 +12,6 @@
 import numpy as np
 from time import time
 
-
 nits = 100
 Ns, D = [10000, 100000, 1000000], 3
 

benchmarks/TVM.py

@@ -41,7 +41,6 @@
 import numpy as np
 from time import time
 
-
 # Global declarations of environment.
 tgt_host = "llvm"
 tgt = "cuda"

doc/conf.py

@@ -108,6 +108,7 @@ def find_source():
     # Add patterns
     # 'filename_pattern': r'../pykeops/pykeops/tutorials/*',
     "ignore_pattern": r"__init__\.py|benchmark_utils\.py|dataset_utils\.py",
+    "parallel": int(os.environ.get("SPHINX_GALLERY_JOBS", "1")),
 }
 
 # Generate the API documentation when building

doc/python/api/common/Config.rst

@@ -0,0 +1,19 @@
+Configuration
+-------------
+
+This section documents lower-level KeOps configuration helpers that are useful
+when troubleshooting PyKeOps installations.
+
+.. rubric:: Summary
+
+.. currentmodule:: keopscore.config.Cuda
+
+.. autosummary::
+    :nosignatures:
+
+    CudaConfig
+
+.. rubric:: Syntax
+
+.. autoclass:: CudaConfig
+    :members:

doc/python/api/common/index.rst

@@ -6,5 +6,6 @@ Common Python API
 .. toctree::
 
     GenericLazyTensor
+    Config
     Utils