xprobe/justfile at master · itdevwu/xprobe · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
set shell := ["bash", "-cu"]

cuda_smoke_image := "nvcr.io/nvidia/cuda:13.3.0-base-ubuntu24.04@sha256:bcf7d05f0b13b9bbb86d9a4cd039d331894b8f1145ad009d1af75023bcd1dc5c"
cuda12_devel_image := "nvidia/cuda:12.9.1-devel-ubuntu24.04@sha256:020bc241a628776338f4d4053fed4c38f6f7f3d7eb5919fecb8de313bb8ba47c"
cuda12_compat_build_image := "nvidia/cuda:12.9.1-devel-ubuntu22.04@sha256:bd4e2680a261c212f1e2fea241606f71497dc67a417f73175d794ec8212b5ba8"
cuda12_min_devel_image := "nvidia/cuda:12.0.1-devel-ubuntu22.04@sha256:0632323ec456b33654d489f3ddd336f3b3ea1c87e6421a91a37f6768e659f08c"
cuda13_devel_image := "nvcr.io/nvidia/cuda:13.3.0-devel-ubuntu24.04@sha256:69e9e39eb8fe2cda271654a0f5eac2f1bb946b2fb9c460eb19c7c3c155f4e64e"

default:
    @just --list

build:
    cargo build --workspace
    cmake -S . -B build -G Ninja
    cmake --build build

test: build
    cargo test --workspace
    ctest --test-dir build --output-on-failure
    python3 tests/agent-contract/test_contract.py target/debug/xprobe
    python3 tests/agent-contract/test_trace_analysis.py
    python3 tests/agent-contract/test_multi_process_workflow.py

test-agent-contract: build
    python3 tests/agent-contract/test_contract.py target/debug/xprobe
    python3 tests/agent-contract/test_trace_analysis.py
    python3 tests/agent-contract/test_multi_process_workflow.py

test-skill-install:
    tests/agent-contract/test_skill_install.sh

test-install:
    sh -n install.sh tests/install/test_install.sh
    bash -n scripts/check-glibc-ceiling.sh tests/install/test_glibc_ceiling.sh
    tests/install/test_glibc_ceiling.sh
    tests/install/test_install.sh

test-bpf:
    cmake -S . -B build -G Ninja -DXPROBE_BUILD_BPF=ON
    cmake --build build --target xprobe-bpf

test-bpf-live: build
    python3 tests/integration/test_uprobe.py "{{cuda_smoke_image}}"
    python3 tests/integration/test_linux.py "{{cuda_smoke_image}}"

test-cupti:
    cmake -S . -B build -G Ninja -DXPROBE_BUILD_CUPTI=ON
    cmake --build build --target xprobe-cupti-smoke
    ctest --test-dir build --output-on-failure -R cupti

test-cupti-live: build
    python3 tests/integration/test_cupti.py "{{cuda13_devel_image}}" target/debug/xprobe

test-cupti-live-cuda12: build
    python3 tests/integration/test_cupti.py "{{cuda12_devel_image}}" target/debug/xprobe

test-cupti-live-cuda12-min: build
    python3 tests/integration/test_cuda12_compat.py "{{cuda12_compat_build_image}}" "{{cuda12_min_devel_image}}" target/debug/xprobe

test-injection-live: build
    python3 tests/integration/test_inject.py "{{cuda13_devel_image}}"

test-injection-live-cuda12: build
    python3 tests/integration/test_inject.py "{{cuda12_devel_image}}"

test-multisource-live: build
    python3 tests/integration/test_multisource.py "{{cuda13_devel_image}}" target/debug/xprobe

test-multisource-live-cuda12: build
    python3 tests/integration/test_multisource.py "{{cuda12_devel_image}}" target/debug/xprobe

test-pytorch-symbols: build
    python3 tests/integration/test_pytorch_symbols.py --python "${PYTORCH_PYTHON:?set PYTORCH_PYTHON to a Python with torch}"

test-pytorch-live: build
    python3 tests/integration/test_pytorch.py "{{cuda_smoke_image}}" "${PYTORCH_ENV:?set PYTORCH_ENV to a Mamba environment with torch}"

test-pytorch-cuda-live: build
    python3 tests/integration/test_pytorch_cuda.py --image "{{cuda12_devel_image}}" --pytorch-env "${PYTORCH_ENV:?set PYTORCH_ENV to a Mamba environment with torch}"

test-nvtx-live: build
    python3 tests/integration/test_nvtx.py --image "{{cuda13_devel_image}}"

test-nvtx-live-cuda12: build
    python3 tests/integration/test_nvtx.py --image "{{cuda12_devel_image}}"

benchmark-gpu:
    python3 benchmarks/cuda-callback/run.py "{{cuda13_devel_image}}"

benchmark-aggregate:
    python3 benchmarks/cuda-aggregate/run.py "{{cuda13_devel_image}}"

benchmark-multiprocess:
    python3 benchmarks/cuda-multiprocess/run.py "{{cuda13_devel_image}}"

fmt:
    cargo fmt --all

fmt-check:
    cargo fmt --all --check

lint:
    cargo clippy --workspace --all-targets -- -D warnings

schemas:
    cargo run --package xprobe-protocol --bin generate-schemas

package:
    scripts/package-release.sh

gpu-smoke:
    docker run --rm --runtime=nvidia --gpus all {{cuda_smoke_image}} nvidia-smi --query-gpu=name,driver_version,compute_cap --format=csv,noheader