feat(lower-tirx): align NVIDIA IKET profiling with the official ABI - #20085
Open
spectrometerHBH wants to merge 2 commits into
Open
feat(lower-tirx): align NVIDIA IKET profiling with the official ABI#20085spectrometerHBH wants to merge 2 commits into
spectrometerHBH wants to merge 2 commits into
Conversation
Contributor
|
Caution The consumer version of Gemini Code Assist on GitHub has been sunset. All code review activity has officially ceased. |
Lock the official integration to CUTLASS DSL 4.6.0 with NVRTC 13.2 and refresh its oracle. Report unproven warp convergence as an advisory warning and handle nested loop control independently.
Implement payload-aware NativeDump and ExtendedNativeDump metadata and event lowering from the locked CUTLASS 4.6.0 oracle. Enable lowering only for validated injected children and keep the pass focused on ABI validation and mechanical IR rewriting without token lifetime, stack balance, or warp convergence proofs.
spectrometerHBH
force-pushed
the
agent/official-iket-profiling
branch
from
August 3, 2026 21:09
a3bfea5 to
6dc4586
Compare
spectrometerHBH
marked this pull request as ready for review
August 3, 2026 22:45
Contributor
|
Caution The consumer version of Gemini Code Assist on GitHub has been sunset. All code review activity has officially ceased. |
tlopex
approved these changes
Aug 4, 2026
Contributor
Author
|
@tvm-bot rerun |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
run-iketchild processesLowerIketfocused on ABI/schema validation and mechanical IR rewriting by removing token-lifetime, stack-balance, and warp-convergence proofsMotivation
The previous lowering covered only no-payload NativeDump instrumentation and attempted to prove runtime control-flow properties that are the caller's and IKET runtime's responsibility. Those analyses could reject valid programs or produce advisory diagnostics without improving the determinism of metadata encoding.
This change aligns the frontend and lowering with the locked official ABI, including payload layouts and ExtendedNativeDump IDs, while retaining compile-time checks required to encode declarations and records safely.
Impact
IKET users can attach supported scalar payloads to marks and range endpoints. Validated injected children can use ordinary TIRx JIT compilation while normal compilation continues to strip frontend-only annotations. Event-name, intrinsic, payload-type,
RangeToken, schema, architecture, declaration-count, and metadata-conflict checks remain enforced; range lifetime, push/pop balance, and warp convergence are left to callers and the runtime.Validation
CPATH=/usr/local/cuda/targets/x86_64-linux/include/cccl cmake --build build --parallelpython -m pytest tests/python/tirx/iket/test_iket_profiler.py tests/python/tirx/iket/test_iket_orchestration.py -q(68 passed, 2 skipped)tirx-kernels/tests/test_megakernel_iket.pycanary (9 passed, 1 skipped)pre-commit run --from-ref upstream/main --to-ref HEADgit diff --check upstream/main...HEAD