Skip to content

Commit 2e2f455

Browse files
rparolinrparolin
committed
Adding a global metapackage level pixi.tom to run all the pytest suites with the same virtual environment
1 parent 87b49b7 commit 2e2f455

2 files changed

Lines changed: 146 additions & 0 deletions

File tree

‎pixi.lock‎

Lines changed: 51 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎pixi.toml‎

Lines changed: 95 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,95 @@
1+
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
#
3+
# SPDX-License-Identifier: Apache-2.0
4+
5+
[workspace]
6+
channels = ["conda-forge"]
7+
platforms = ["linux-64", "linux-aarch64", "win-64"]
8+
preview = ["pixi-build"]
9+
10+
[dependencies]
11+
12+
[feature.cu12.system-requirements]
13+
cuda = "12"
14+
15+
[feature.cu12.dependencies]
16+
cuda-version = ">=12.0,<13"
17+
18+
[feature.cu13.system-requirements]
19+
cuda = "13"
20+
21+
[feature.cu13.dependencies]
22+
cuda-version = ">=13.1,<14"
23+
24+
[environments]
25+
cu12 = { features = ["cu12"], solve-group = "cu12" }
26+
cu13 = { features = ["cu13"], solve-group = "cu13" }
27+
28+
# Install tasks: Set up all sub-package environments
29+
# Usage: pixi run install (default) or pixi run -e cu12 install or pixi run -e cu13 install
30+
# Detects environment from CONDA_DEFAULT_ENV (set by pixi when using -e flag)
31+
[target.linux.tasks.install-pathfinder]
32+
cmd = ["bash", "-c", "ENV=\"\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu12\"* ]] && ENV=\"cu12\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu13\"* ]] && ENV=\"cu13\"; cd cuda_pathfinder && pixi ${ENV:+--environment $ENV} install"]
33+
34+
[target.linux.tasks.install-bindings]
35+
cmd = ["bash", "-c", "ENV=\"\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu12\"* ]] && ENV=\"cu12\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu13\"* ]] && ENV=\"cu13\"; cd cuda_bindings && pixi ${ENV:+--environment $ENV} install"]
36+
37+
[target.linux.tasks.install-core]
38+
cmd = ["bash", "-c", "ENV=\"\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu12\"* ]] && ENV=\"cu12\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu13\"* ]] && ENV=\"cu13\"; cd cuda_core && pixi ${ENV:+--environment $ENV} install"]
39+
40+
[target.linux.tasks.install]
41+
cmd = [
42+
"bash", "-c",
43+
"set -e; ENV=\"\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu12\"* ]] && ENV=\"cu12\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu13\"* ]] && ENV=\"cu13\"; cd cuda_pathfinder && pixi ${ENV:+--environment $ENV} install && cd ../cuda_bindings && pixi ${ENV:+--environment $ENV} install && cd ../cuda_core && pixi ${ENV:+--environment $ENV} install"
44+
]
45+
46+
47+
# Test tasks: Run tests (assumes environments are already installed)
48+
# Order: pathfinder -> bindings -> core (matching scripts/run_tests.sh)
49+
# Each sub-package has its own pixi.toml and environment
50+
# Usage: pixi run test (default) or pixi run -e cu12 test or pixi run -e cu13 test
51+
[target.linux.tasks.test-pathfinder]
52+
cmd = ["bash", "-c", "ENV=\"\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu12\"* ]] && ENV=\"cu12\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu13\"* ]] && ENV=\"cu13\"; cd cuda_pathfinder && if [ -n \"$ENV\" ]; then pixi run --environment \"$ENV\" test; else pixi run test; fi"]
53+
54+
[target.linux.tasks.test-bindings]
55+
cmd = ["bash", "-c", "ENV=\"\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu12\"* ]] && ENV=\"cu12\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu13\"* ]] && ENV=\"cu13\"; cd cuda_bindings && if [ -n \"$ENV\" ]; then pixi run --environment \"$ENV\" test; else pixi run test; fi"]
56+
57+
[target.linux.tasks.test-core]
58+
cmd = ["bash", "-c", "ENV=\"\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu12\"* ]] && ENV=\"cu12\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu13\"* ]] && ENV=\"cu13\"; cd cuda_core && if [ -n \"$ENV\" ]; then pixi run --environment \"$ENV\" test; else pixi run test; fi"]
59+
60+
[target.linux.tasks.test]
61+
cmd = [
62+
"bash", "-c",
63+
"set -e; ENV=\"\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu12\"* ]] && ENV=\"cu12\"; [[ \"${CONDA_DEFAULT_ENV:-}\" == *\"cu13\"* ]] && ENV=\"cu13\"; cd cuda_pathfinder && (if [ -n \"$ENV\" ]; then pixi run --environment \"$ENV\" test; else pixi run test; fi) && cd ../cuda_bindings && (if [ -n \"$ENV\" ]; then pixi run --environment \"$ENV\" test; else pixi run test; fi) && cd ../cuda_core && (if [ -n \"$ENV\" ]; then pixi run --environment \"$ENV\" test; else pixi run test; fi)"
64+
]
65+
66+
67+
[target.win-64.tasks.install-pathfinder]
68+
cmd = ["cmd", "/c", "set ENV= && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu12\" >nul && set ENV=cu12 && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu13\" >nul && set ENV=cu13 && cd cuda_pathfinder && (if defined ENV (pixi --environment %ENV% install) else (pixi install))"]
69+
70+
[target.win-64.tasks.install-bindings]
71+
cmd = ["cmd", "/c", "set ENV= && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu12\" >nul && set ENV=cu12 && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu13\" >nul && set ENV=cu13 && cd cuda_bindings && (if defined ENV (pixi --environment %ENV% install) else (pixi install))"]
72+
73+
[target.win-64.tasks.install-core]
74+
cmd = ["cmd", "/c", "set ENV= && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu12\" >nul && set ENV=cu12 && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu13\" >nul && set ENV=cu13 && cd cuda_core && (if defined ENV (pixi --environment %ENV% install) else (pixi install))"]
75+
76+
[target.win-64.tasks.install]
77+
cmd = [
78+
"cmd", "/c",
79+
"set ENV= && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu12\" >nul && set ENV=cu12 && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu13\" >nul && set ENV=cu13 && cd cuda_pathfinder && (if defined ENV (pixi --environment %ENV% install) else (pixi install)) && cd ..\\cuda_bindings && (if defined ENV (pixi --environment %ENV% install) else (pixi install)) && cd ..\\cuda_core && (if defined ENV (pixi --environment %ENV% install) else (pixi install))"
80+
]
81+
82+
[target.win-64.tasks.test-pathfinder]
83+
cmd = ["cmd", "/c", "set ENV= && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu12\" >nul && set ENV=cu12 && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu13\" >nul && set ENV=cu13 && cd cuda_pathfinder && (if defined ENV (pixi run --environment %ENV% test) else (pixi run test))"]
84+
85+
[target.win-64.tasks.test-bindings]
86+
cmd = ["cmd", "/c", "set ENV= && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu12\" >nul && set ENV=cu12 && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu13\" >nul && set ENV=cu13 && cd cuda_bindings && (if defined ENV (pixi run --environment %ENV% test) else (pixi run test))"]
87+
88+
[target.win-64.tasks.test-core]
89+
cmd = ["cmd", "/c", "set ENV= && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu12\" >nul && set ENV=cu12 && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu13\" >nul && set ENV=cu13 && cd cuda_core && (if defined ENV (pixi run --environment %ENV% test) else (pixi run test))"]
90+
91+
[target.win-64.tasks.test]
92+
cmd = [
93+
"cmd", "/c",
94+
"set ENV= && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu12\" >nul && set ENV=cu12 && echo %CONDA_DEFAULT_ENV% | findstr /C:\"cu13\" >nul && set ENV=cu13 && cd cuda_pathfinder && (if defined ENV (pixi run --environment %ENV% test) else (pixi run test)) && cd ..\\cuda_bindings && (if defined ENV (pixi run --environment %ENV% test) else (pixi run test)) && cd ..\\cuda_core && (if defined ENV (pixi run --environment %ENV% test) else (pixi run test))"
95+
]

0 commit comments

Comments
 (0)