mirror of https://github.com/llvm/torch-mlir
8f52f5a4ed
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> |
||
|---|---|---|
| .. | ||
| fx_importer | ||
| onnx_importer | ||
| compile.py | ||
| lit.local.cfg | ||