Skip to content

Support Mamba direct conv1d quant export#1716

Draft
mxinO wants to merge 2 commits into
mainfrom
mxin/fix-export-megatron
Draft

Support Mamba direct conv1d quant export#1716
mxinO wants to merge 2 commits into
mainfrom
mxin/fix-export-megatron

Preserve Mamba Conv1d quantizer state for export

d9b8718
Select commit
Loading
Failed to load commit list.
Codecov / codecov/project succeeded Jun 14, 2026 in 1s

76.98% (-0.15%) compared to 9f37fe1

View this Pull Request on Codecov

76.98% (-0.15%) compared to 9f37fe1

Details

Codecov Report

❌ Patch coverage is 7.01754% with 106 lines in your changes missing coverage. Please review.
✅ Project coverage is 76.98%. Comparing base (9f37fe1) to head (d9b8718).
⚠️ Report is 1 commits behind head on main.

Files with missing lines Patch % Lines
modelopt/torch/quantization/plugins/megatron.py 1.17% 84 Missing ⚠️
modelopt/torch/export/unified_export_megatron.py 16.00% 21 Missing ⚠️
modelopt/torch/quantization/backends/nvfp4_gemm.py 0.00% 1 Missing ⚠️
Additional details and impacted files 
@@            Coverage Diff             @@
##             main    #1716      +/-   ##
==========================================
- Coverage   77.12%   76.98%   -0.15%     
==========================================
  Files         511      511              
  Lines       56236    56347     +111     
==========================================
+ Hits        43374    43380       +6     
- Misses      12862    12967     +105

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.
View more details on Codecov