[CI] Add workflow verifying build against minimum supported PyTorch and JAX - #3396

Open

pggPL wants to merge 18 commits into

NVIDIA:mainfrom

pggPL:pytorch21_build_ci

Commits on Aug 18, 2026

Add CI workflow building TE against torch 2.1
pggPL
committed
3875204
Remove MAX_JOBS=1 from pytorch 2.1 build workflow
pggPL
committed
fba3898
Use MAX_JOBS=2 in pytorch 2.1 workflow, unbounded build OOMs the runner
pggPL
committed
3b600dd
Point TE build at apt cudnn9, torch 2.1 pip deps shadow it with cudnn 8.9
pggPL
committed
dce613c
Avoid passing std::optional to at::get_generator_or_default, torch 2.1 needs c10::optional
pggPL
committed
061834f
Gate NCCL EP in torch extension on nccl_ep_enabled, matching common CMake
pggPL
committed
32a28ef
Pass comm streams as raw cudaStream_t handles, torch 2.1 pybind lacks c10::Stream caster
pggPL
committed
04b4b0f
Revert "Pass comm streams as raw cudaStream_t handles, torch 2.1 pybind lacks c10::Stream caster"
pggPL
committed
4a3ad66
Revert "Avoid passing std::optional to at::get_generator_or_default, torch 2.1 needs c10::optional"
pggPL
committed
2087bd0
Retarget CI workflow to torch 2.8
pggPL
committed
690701a
Rename workflow to Minimum supported PyTorch, parametrize torch version
pggPL
committed
7b7c6be
Add Minimum supported JAX job, rename workflow to minimum_versions
pggPL
committed
b86e86c
Add packaging to min-jax job deps
pggPL
committed
12f725f
Skip NCCL EP in torch extension when torch lacks symm-mem headers; build min-torch job on sm90
pggPL
committed
f094de7
Fix min-versions jobs: pydantic for jax import, libcuda stub for sm90 import
pggPL
committed
98324dd

Commits on Aug 19, 2026

Extract torch symm-mem EP gate into nccl_ep_supported_by_torch helper
pggPL
committed
f034956
Tidy nccl_ep_supported_by_torch
pggPL
committed
63311ef
Compile out EP zero-copy path when torch lacks symm-mem headers
pggPL
committed
e64a6a1