Add dependencies.yaml.

.pre-commit-config.yaml

@@ -8,6 +8,11 @@ repos:
     rev: v16.0.6
     hooks:
       - id: clang-format
+  - repo: https://github.com/rapidsai/dependency-file-generator
+    rev: v1.13.11
+    hooks:
+      - id: rapids-dependency-file-generator
+        args: ["--clean"]
   - repo: https://github.com/astral-sh/ruff-pre-commit
     rev: v0.4.6
     hooks:

dependencies.yaml

@@ -0,0 +1,109 @@
+# Dependency list for https://github.com/rapidsai/dependency-file-generator
+files:
+  test:
+    output: none
+    includes:
+      - cuda_version
+      - py_version
+      - cuda
+      - test
+  checks:
+    output: none
+    includes:
+      - checks
+      - py_version
+  py_build:
+    output: pyproject
+    pyproject_dir: .
+    extras:
+      table: build-system
+    includes:
+      - build
+  py_run:
+    output: pyproject
+    pyproject_dir: .
+    extras:
+      table: project
+    includes:
+      - run
+  py_optional_test:
+    output: pyproject
+    pyproject_dir: .
+    extras:
+      table: project.optional-dependencies
+      key: test
+    includes:
+      - test
+channels:
+  - rapidsai
+  - rapidsai-nightly
+  - conda-forge
+dependencies:
+  build:
+    common:
+      - output_types: [requirements, pyproject]
+        packages:
+          - scikit-build-core[pyproject]>=0.7.0
+  checks:
+    common:
+      - output_types: [conda, requirements]
+        packages:
+          - pre-commit
+  cuda_version:
+    specific:
+      - output_types: conda
+        matrices:
+          - matrix:
+              cuda: "12.0"
+            packages:
+              - cuda-version=12.0
+  cuda:
+    specific:
+      - output_types: conda
+        matrices:
+          - matrix:
+              cuda: "12.*"
+            packages:
+              - cuda-nvcc
+              - cuda-nvrtc
+  py_version:
+    specific:
+      - output_types: conda
+        matrices:
+          - matrix:
+              py: "3.9"
+            packages:
+              - python=3.9
+          - matrix:
+              py: "3.10"
+            packages:
+              - python=3.10
+          - matrix:
+              py: "3.11"
+            packages:
+              - python=3.11
+          - matrix:
+              py: "3.12"
+            packages:
+              - python=3.12
+          - matrix:
+            packages:
+              - python>=3.9,<3.13
+  run:
+    common:
+      - output_types: [conda, requirements, pyproject]
+        packages: []
+  test:
+    common:
+      - output_types: conda
+        packages:
+          - c-compiler
+          - cxx-compiler
+          - make
+      - output_types: [conda, requirements, pyproject]
+        packages:
+          - cuda-python
+          - numba>=0.58
+          - psutil
+          - pytest
+          - pytest-cov

pyproject.toml

@@ -1,3 +1,5 @@
+# Copyright (c) 2023-2024, NVIDIA CORPORATION. All rights reserved.
+
 [tool.scikit-build]
 cmake.minimum-version = "3.26.4"
 cmake.verbose = true
@@ -6,7 +8,9 @@ build-dir = "build/{wheel_tag}"
 wheel.packages = ["pynvjitlink"]
 
 [build-system]
-requires = ["scikit-build-core"]
+requires = [
+    "scikit-build-core[pyproject]>=0.7.0",
+] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.
 build-backend = "scikit_build_core.build"
 
 [project]
@@ -19,13 +23,25 @@ authors = [
 ]
 license = { text = "Apache 2.0" }
 requires-python = ">=3.9"
+dependencies = [
+] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [project.urls]
 Homepage = "https://rapids.ai/"
 Documentation = "https://github.com/rapidsai/pynvjitlink/blob/main/README.md"
 Repository = "https://github.com/rapidsai/pynvjitlink"
 License = "https://github.com/rapidsai/pynvjitlink/blob/main/LICENSE"
 
+
+[project.optional-dependencies]
+test = [
+    "cuda-python",
+    "numba>=0.58",
+    "psutil",
+    "pytest",
+    "pytest-cov",
+] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.
+
 [tool.setuptools]
 license-files = ["LICENSE"]