Feature/nano v2 offline modelopt fp8 and nvfp4 (#12018)

Signed-off-by: Netanel Haber <58652339+netanel-haber@users.noreply.github.com>
2025-10-23 21:16:46 +03:00
parent a4b637d87a
commit a98496834b
11 changed files with 213 additions and 140 deletions
--- a/test/srt/layers/attention/mamba/test_causal_conv1d.py
+++ b/test/srt/layers/attention/mamba/test_causal_conv1d.py
@@ -373,3 +373,7 @@ def test_causal_conv1d_varlen(
    )
    unpadded_out = out[:, : out_ref_tensor.shape[-1]]
    assert torch.allclose(unpadded_out, out_ref_tensor, rtol=rtol, atol=atol)
+
+
+if __name__ == "__main__":
+    pytest.main([__file__])