torch-mlir/test/python/fx_importer/v2.3
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
..
auto_functionalized.py [fx] Implement auto_functionalized higher order op. (#3063) 2024-03-26 17:06:05 -07:00
lit.local.cfg [NFC reformat] Run pre-commit on all files and format misc. 2024-04-27 14:08:09 -07:00
mutation_import.py [Fx Importer] fix mutation importer with non persistent buffer (#3798) 2024-11-01 02:20:32 +08:00
special_forms_test.py [fx] Fix type determination for multi-return ops and static `None` returns. (#3081) 2024-04-01 09:39:38 -07:00
types_test.py build: Update Roll PyTorch version (#3548) 2024-07-19 21:38:57 +05:30