crash in rewriter/broadcast_to_matmul.py (proposed fix) about onnxscript HOT 3 CLOSED

import torch
from nemo.core.classes import typecheck
from nemo.utils.export_utils import wrap_forward_method
from nemo.collections.asr.models import EncDecCTCModelBPE
import onnxscript

model = EncDecCTCModelBPE.from_pretrained(model_name="stt_en_conformer_ctc_small")

model = model.eval()
wrap_forward_method(model)
model._prepare_for_export()

typecheck.set_typecheck_enabled(enabled=False)

with torch.no_grad():
    input_example = model.input_module.input_example(max_batch=4, max_dim=256)
    # This has to have min=2                                                                                                              
    batch = torch.export.Dim("batch", min=2, max=64)
    dynamic_shapes={'audio_signal': { 0: batch,
                                      # 2: torch.export.Dim("audio_signal__2")                                                            
                              },
                    'length': {0: batch}
                    }

    ex_model = torch.export.export(
        model, tuple(input_example),
        dynamic_shapes=dynamic_shapes,
        strict=False
    )

    ex_model = ex_model.run_decompositions()
    ex_module=ex_model.module()

    print("Running torch.onnx.dynamo_export ...")
    options = torch.onnx.ExportOptions(dynamic_shapes=True)
    ex = torch.onnx.dynamo_export(ex_module, *input_example, export_options=options)
    ex.save("ctc.onnx")