无法执行GraphModule中的torch__dynamo_utils_wrapped_prod函数

[fangfangssj]

transformers库中的tts系列模型，运行第一次提取出来的计算图GraphModule（dynamo在跟踪时生成的）会报错，无法执行torch__dynamo_utils_wrapped_prod函数

Traceback (most recent call last):
  File "<frozen runpy>", line 198, in _run_module_as_main
  File "<frozen runpy>", line 88, in _run_code
  File "/root/GraphNet/graph_net/torch/single_device_runner.py", line 107, in <module>
    main(args=args)
  File "/root/GraphNet/graph_net/torch/single_device_runner.py", line 72, in main
    y = model(**state_dict)[0]
        ^^^^^^^^^^^^^^^^^^^
  File "/root/miniconda3/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1751, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/root/miniconda3/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1762, in _call_impl
    return forward_call(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/root/graphnet_work/facebook/mms-tts-ind/model.py", line 372, in forward
    wrapped_prod = torch__dynamo_utils_wrapped_prod([8, 8, 2, 2])
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
NameError: name 'torch__dynamo_utils_wrapped_prod' is not defined