[mgohil-png]

What does this PR do?

Type of change: new feature

Simplify the supported TopK/Cast/Reshape/Tile/GatherElements chain, handle eligible integer Unsqueeze inputs, and provide standalone InstanceNormalization channel-layout compatibility. Preserve unsupported/shared patterns and register the ordered TopK and Unsqueeze stages.

Part 15/17. Depends on part 14; merge from the bottom of the stack upward.

Usage

from modelopt.onnx.graph_surgery.dla_transforms.dla_topk import _apply_topk

# model is an onnx.ModelProto with the supported input pattern.
model = _apply_topk(model)

This is an internal pass; the public pipeline is registered in part 16.

Testing

• Validated on this branch's current commit on Windows CPU with Python 3.10 and ONNX 1.21.0: 7 passed, 1 warning.
• TopK pattern matching and outputs, shared/unsupported patterns, and InstanceNormalization layout behavior.
• Repository pre-commit hooks passed for this slice; the commit is signed off and its SSH signature was verified.

python -m pytest tests/unit/onnx/graph_surgery/dla_transforms/test_dla_fix_instancenorm_channel_mismatch.py tests/unit/onnx/graph_surgery/dla_transforms/test_dla_topk.py --noconftest -o addopts= -q --tb=short --durations=5

Tests ran with Hugging Face offline mode and without the repository-wide conftest/addopts; test-module fixtures still ran. GPU execution and DLA compilation were not run. The final stacked branch also passed the combined focused suite: 317 passed, 2 warnings.

Before your PR is "Ready for review"

• Backward compatible: yes; this slice adds DLA-specific functionality.
• Third-party code/new runtime dependencies: no new runtime dependency; contribution and security guidance reviewed.
• Necessary tests: included for this slice, with focused synthetic fixtures.
• Changelog: public feature entry follows in part 16.
• Claude approval: pending for the current commit. Submitted as a draft.

Additional Information

Base branch: dla/review-14-dla-boolean-and-comparison-rewrites. Review this slice against that branch; subsequent slices are reviewed separately.

Merge order: PR1 (merged) → Part 2/17 → Part 3/17 → Part 4/17 → Part 5/17 → Part 6/17 → Part 7/17 → Part 8/17 → Part 9/17 → Part 10/17 → Part 11/17 → Part 12/17 → Part 13/17 → Part 14/17 → Part 15/17 → Part 16/17 → Part 17/17

[copy-pr-bot]

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

[coderabbitai]

Important

Draft PR not reviewed

Draft PRs are not automatically reviewed by default.

• Trigger a manual review

To automatically review draft PRs, update your CodeRabbit configuration:

reviews:
  auto_review:
    drafts: true

• Autofix · Keep fixing CodeRabbit findings and required CI, and resolving merge conflicts

_{Comment @coderabbitai help to get the list of available commands.}

[mgohil-png]

@coderabbitai full review

[coderabbitai]

⚠️ Action not completed

Review rate limited.

[github-actions]

PR Preview Action v1.8.1
🚀 View preview at https://NVIDIA.github.io/Model-Optimizer/pr-preview/pr-2967/
Built to branch gh-pages at 2026-10-23 17:34 UTC. Preview will be ready when the GitHub Pages deployment is complete.