Add pre-commit CI job and replace artifact-based generated-file checks

The check_generated_files.py pre-commit hook regenerates exports.h,
version.h, and __init__.pyi from Python source without needing build
artifacts. Add it (along with a version-consistency hook) to
.pre-commit-config.yaml and run it in a new lightweight pre-commit CI
job in both GitHub Actions and GitLab CI.

This replaces the artifact-dependent git-diff checks:
- GitHub: remove check-generated-files job, remove stubs check from
  check-build-docs-output
- GitLab: slim check-generated-files down to docs-only, rename to
  check-generated-docs

Move pre-commit hook scripts to tools/pre-commit-hooks/ for
discoverability.

Signed-off-by: Eric Shi <ershi@nvidia.com>

Author: shi-eric

.github/workflows/ci.yml

@@ -243,6 +243,20 @@ jobs:
             docs/api_reference
             docs/language_reference
 
+  pre-commit:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+      - name: Install uv
+        uses: astral-sh/setup-uv@d4b2f3b6ecc6e67c4457f6d3e41ec42d3d0fcb86
+        with:
+          version: "0.8.23"
+      - name: Set up Python
+        run: uv python install
+      - name: Run pre-commit
+        run: uvx pre-commit run --all-files --show-diff-on-failure
+
   check-build-docs-output:
     runs-on: ubuntu-latest
     needs: build-docs
@@ -260,40 +274,6 @@ jobs:
         run: >
           git diff --exit-code docs/api_reference docs/language_reference ||
           (echo "Please run build_docs.py (or download from $ARTIFACT_URL) and add docs/api_reference and docs/language_reference to your pull request." && false)
-      - name: Check stubs
-        env:
-          ARTIFACT_URL: ${{ needs.build-docs.outputs.artifact-url }}
-        run: >
-          git diff --exit-code warp/__init__.pyi ||
-          (echo "Please run build_docs.py (or download from $ARTIFACT_URL) and add warp/__init__.pyi to your pull request." && false)
-
-  check-generated-files:
-    runs-on: ubuntu-latest
-    needs: build-warp
-    steps:
-      - name: Checkout repository
-        uses: actions/checkout@v4
-      - name: Install uv
-        uses: astral-sh/setup-uv@d4b2f3b6ecc6e67c4457f6d3e41ec42d3d0fcb86
-        with:
-          version: "0.8.23"
-      - name: Set up Python
-        run: uv python install
-      - name: Download native headers
-        uses: actions/download-artifact@v4
-        with:
-          name: native-headers-ubuntu-x86_64
-          path: warp/native/
-      - name: Check version consistency
-        run: uv run --no-project tools/ci/publishing/check_version_consistency.py --verbose
-      - name: Check exports.h is committed
-        run: >
-          git diff --exit-code warp/native/exports.h ||
-          (echo "Please run build_lib.py and add warp/native/exports.h to your pull request." && false)
-      - name: Check version.h is committed
-        run: >
-          git diff --exit-code warp/native/version.h ||
-          (echo "Please run build_lib.py and add warp/native/version.h to your pull request." && false)
 
   check-unnamespaced-symbols:
     runs-on: ubuntu-latest

.gitlab-ci.yml

@@ -958,13 +958,12 @@ merge request docs:
   after_script:
     - echo "View the website at https://$CI_PROJECT_ROOT_NAMESPACE.$CI_PAGES_DOMAIN/-/$CI_PROJECT_NAME/-/jobs/$CI_JOB_ID/artifacts/public/index.html"
 
-# Merge requests: Ensure that exports.h, __init__.pyi, and docs reference files have been
+# Merge requests: Ensure that docs reference files generated by Sphinx have been
 # manually added to the MR if they are changed
-check generated files:
+check generated docs:
   stage: deploy
   image: $UV_TRIXIE_IMAGE
   needs:
-    - job: linux-x86_64 build
     - job: merge request docs
       optional: true
   extends:
@@ -976,25 +975,12 @@ check generated files:
     when: on_failure
     expose_as: "Generated source files"
     paths:
-      - warp/native/exports.h
-      - warp/native/version.h
-      - warp/__init__.pyi
       - docs/api_reference
       - docs/language_reference
   script:
-    # Check version consistency across VERSION.md, config.py, and version.h
-    - uv run tools/ci/publishing/check_version_consistency.py --verbose
-    # Check that generated documentation files are committed
-    - >
-      git diff --exit-code warp/__init__.pyi docs/api_reference docs/language_reference ||
-      (echo "Please run build_docs.py (or download from $CI_JOB_URL/artifacts/browse) and add modified files to your merge request." && false)
-    # Check that generated native files are committed
-    - >
-      git diff --exit-code warp/native/exports.h ||
-      (echo "Please run build_lib.py (or download from $CI_JOB_URL/artifacts/browse) and add warp/native/exports.h to your merge request." && false)
     - >
-      git diff --exit-code warp/native/version.h ||
-      (echo "Please run build_lib.py (or download from $CI_JOB_URL/artifacts/browse) and add warp/native/version.h to your merge request." && false)
+      git diff --exit-code docs/api_reference docs/language_reference ||
+      (echo "Please run build_docs.py (or download from $CI_JOB_URL/artifacts/browse) and add docs/api_reference and docs/language_reference to your merge request." && false)
 
 # Build documentation and publish on GitLab
 # This only runs in the default branch pipeline. The "pages" name is special for GitLab.

.pre-commit-config.yaml

@@ -5,12 +5,28 @@ ci:
   autoupdate_branch: ""
   autoupdate_commit_msg: "[pre-commit.ci] pre-commit autoupdate"
   autoupdate_schedule: quarterly
-  skip: []
+  skip: [warp-check-generated-files]
   submodules: false
 
 # See https://pre-commit.com for more information
 # See https://pre-commit.com/hooks.html for more hooks
 repos:
+  - repo: local
+    hooks:
+      - id: warp-check-generated-files
+        name: "warp: regenerate generated files"
+        entry: python tools/pre-commit-hooks/check_generated_files.py
+        language: python
+        # NOTE: ruff version must match ruff-pre-commit rev below to avoid formatting drift.
+        additional_dependencies: [numpy, ruff==0.14.10]
+        always_run: true
+        pass_filenames: false
+      - id: warp-check-version-consistency
+        name: "warp: check version consistency"
+        entry: python tools/pre-commit-hooks/check_version_consistency.py --verbose
+        language: python
+        always_run: true
+        pass_filenames: false
   - repo: https://github.com/astral-sh/ruff-pre-commit
     # Ruff version.
     rev: v0.14.10

build_lib.py

@@ -38,7 +38,7 @@
 import build_llvm
 import warp._src.build_dll as build_dll
 import warp.config as config
-from warp._src.context import export_builtins
+from warp._src.generated_files import generate_exports_header_file, generate_version_header
 
 
 def handle_ci_nightly_build(base_path: str) -> str | None:
@@ -92,40 +92,6 @@ def handle_ci_nightly_build(base_path: str) -> str | None:
     return dev_version_string
 
 
-def generate_version_header(base_path: str, version: str) -> None:
-    """Generate version.h with WP_VERSION_STRING macro."""
-    version_header_path = os.path.join(base_path, "warp", "native", "version.h")
-    current_year = datetime.date.today().year
-
-    copyright_notice = f"""/*
- * SPDX-FileCopyrightText: Copyright (c) {current_year} NVIDIA CORPORATION & AFFILIATES. All rights reserved.
- * SPDX-License-Identifier: Apache-2.0
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-"""
-
-    with open(version_header_path, "w") as f:
-        f.write(copyright_notice)
-        f.write("#ifndef WP_VERSION_H\n")
-        f.write("#define WP_VERSION_H\n\n")
-        f.write(f'#define WP_VERSION_STRING "{version}"\n\n')
-        f.write("#endif  // WP_VERSION_H\n")
-
-    print(f"Generated {version_header_path} with version {version}")
-
-
 def find_cuda_sdk() -> str | None:
     # check environment variables
     for env in ["WARP_CUDA_PATH", "CUDA_HOME", "CUDA_PATH"]:
@@ -249,45 +215,6 @@ def lib_name(name: str) -> str:
         return f"{name}.so"
 
 
-def generate_exports_header_file(base_path: str) -> None:
-    """Generates warp/native/exports.h, which lets built-in functions be callable from outside kernels."""
-    export_path = os.path.join(base_path, "warp", "native", "exports.h")
-    os.makedirs(os.path.dirname(export_path), exist_ok=True)
-
-    try:
-        with open(export_path, "w") as f:
-            copyright_notice = """/*
- * SPDX-FileCopyrightText: Copyright (c) 2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
- * SPDX-License-Identifier: Apache-2.0
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-"""
-            f.write(copyright_notice)
-            export_builtins(f)
-
-        print(f"Finished writing {export_path}")
-    except FileNotFoundError:
-        print(f"Error: The file '{export_path}' was not found.")
-    except PermissionError:
-        print(f"Error: Permission denied. Unable to write to '{export_path}'.")
-    except OSError as e:
-        print(f"Error: An OS-related error occurred: {e}")
-    except Exception as e:
-        print(f"An unexpected error occurred: {e}")
-
-
 def main(argv: list[str] | None = None) -> int:
     parser = argparse.ArgumentParser(
         description="Build Warp native libraries with optional CUDA, LLVM, and MathDx support",

tools/pre-commit-hooks/check_generated_files.py

@@ -0,0 +1,70 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Pre-commit hook that regenerates exports.h, version.h, and __init__.pyi.
+
+These files are derived from warp/_src/context.py, warp/config.py, VERSION.md,
+and related source files. When those sources change, the generated files must be
+updated to match. This hook regenerates them in place; the pre-commit framework
+detects any modifications and blocks the commit so the developer can stage them.
+
+Usage (pre-commit hook):
+    Configured in .pre-commit-config.yaml with language: python
+    and additional_dependencies: [numpy, ruff]
+"""
+
+from __future__ import annotations
+
+import os
+import subprocess
+import sys
+
+# Ensure the repo root is on sys.path so warp is importable in the
+# pre-commit isolated virtualenv (which only has additional_dependencies).
+base_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+sys.path.insert(0, base_path)
+
+import warp  # noqa: E402 — populates builtin_functions via builtins.py
+import warp.config  # noqa: E402
+from warp._src.generated_files import export_stubs, generate_exports_header_file, generate_version_header  # noqa: E402
+
+
+def main() -> int:
+    # Regenerate warp/native/exports.h
+    generate_exports_header_file(base_path)
+
+    # Regenerate warp/native/version.h
+    generate_version_header(base_path, warp.config.version)
+
+    # Regenerate warp/__init__.pyi
+    stub_path = os.path.join(base_path, "warp", "__init__.pyi")
+    with open(stub_path, "w", encoding="utf-8") as f:
+        export_stubs(f)
+
+    # Format with ruff so the generated file is stable across hook runs.
+    # export_stubs() emits trailing whitespace and non-canonical formatting;
+    # without this step, ruff-format would modify the file on every commit.
+    # ruff is provided via additional_dependencies in .pre-commit-config.yaml.
+    from ruff.__main__ import find_ruff_bin  # noqa: PLC0415
+
+    ruff = find_ruff_bin()
+    subprocess.run([ruff, "format", stub_path], check=True)
+    subprocess.run([ruff, "check", "--fix", "--unsafe-fixes", stub_path], check=False)
+
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())