torch-mlir/python/torch_mlir/extras
Dixin Zhou 8f52f5a4ed
[Fx Importer] fix mutation importer with non persistent buffer (#3798)
A non-persistent buffer will not be a part of this module’s
`state_dict`. Hence when setting `experimental_support_mutation=True`
and have non-persistent buffer, the current fx importer will fail to
retrieve a value from `state_dict` and produce `torch.constant.none` to
represent the buffer. This fix get value of non-persistent buffer from
the module's `constants`.

---------

Co-authored-by: Dixin Zhou <dzhou@vdi-ahddp-020.dhcp.mathworks.com>
2024-11-01 02:20:32 +08:00
..
fx_decomp_util.py [FxImporter] small fixes for fx importer compatibility issues between different pytorch versions (#3577) 2024-08-01 10:52:41 +08:00
fx_importer.py [Fx Importer] fix mutation importer with non persistent buffer (#3798) 2024-11-01 02:20:32 +08:00
onnx_importer.py Bump Onnx Version to 1.16.1 (#3515) 2024-07-01 22:15:45 +05:30