Skip to content

Add Warp support to the compiled runtime’s ONNX and PT2 backends - #46

Open
malwan-nv wants to merge 3 commits into
malwan-nv/leapp-runtime-pluginsfrom
malwan-nv/warp-runtime-plugin
Open

malwan-nv wants to merge 3 commits into
malwan-nv/leapp-runtime-pluginsfrom
malwan-nv/warp-runtime-plugin

Conversation

@malwan-nv

Copy link
Copy Markdown
Collaborator

This adds Warp support to compiled LEAPP sessions through native adapters for Triton’s ONNX and PT2 backends. It supports CPU and CUDA tensor placement, CUDA stream ordering, noncontiguous Torch inputs, and graph cleanup when models unload. Build improvements add explicit SDK selection, installable libraries, and dependency validation before replacing cached builds. Validation covers CUDA replay, all 12 placement combinations, session lifecycle, and adapter load failures.

Discover installed Warp adapters for Triton ONNX Runtime and NativeRT/PT2
models. Keep ONNX CUDA replay tensors on the GPU and support native Torch
registration, stream ordering, and cleanup when models unload.

Add explicit SDK selection, generic CMake argument forwarding, and install
rules without embedding the absolute Warp library path in dependencies.
Validate SDKs before replacing an existing cached runtime build.

Validation: seven build-helper tests, two native CUDA replay tests, and
two Triton integration tests covering twelve session placements and load
errors. Verified SDK overrides, relocated installs, and formatting.

Signed-off-by: Marc Alwan <malwan@nvidia.com>
@malwan-nv
malwan-nv requested a review from frlai October 23, 2026 22:30
@malwan-nv malwan-nv self-assigned this Oct 23, 2026
@malwan-nv
malwan-nv added this pull request to stack #43 October 23, 2026 22:30
Signed-off-by: Marc Alwan <malwan@nvidia.com>
Default the new engine option in the Warp integration test.

Signed-off-by: Marc Alwan <malwan@nvidia.com>
@malwan-nv
malwan-nv removed this pull request from stack #43 October 26, 2026 18:43
@malwan-nv
malwan-nv added this pull request to stack #47 October 26, 2026 18:44

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant