Unverified Commit 918a9ad7 authored by galagam's avatar galagam Committed by GitHub
Browse files

ONNX export test - fix file name in transformer layer tests (#288)



Same filename is used for all activations, file is getting overridden to last activation type
Signed-off-by: default avatarGal Hubara Agam <ghubaraagam@nvidia.com>
Co-authored-by: default avatarKirthi Shankar Sivamani <ksivamani@nvidia.com>
parent 0816583a
......@@ -1258,7 +1258,7 @@ def test_export_transformer_layer(
fuse_qkv_params_str = "_fused-qkv" if fuse_qkv_params else ""
high_prec_str = dtype2str(precision)
attn_mask_str = get_attn_mask_str(use_mask, attn_mask_type)
fname = f"te.transformer_layer{fp8_str}{attn_mask_str}{fuse_qkv_params_str}{high_prec_str}.onnx"
fname = f"te.transformer_layer{fp8_str}{attn_mask_str}{fuse_qkv_params_str}{high_prec_str}_{activation}.onnx"
model = te.TransformerLayer(
hidden_size,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment