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> |
||
|---|---|---|
| .. | ||
| _mlir_libs | ||
| dialects | ||
| extras | ||
| tools/import_onnx | ||
| compiler_utils.py | ||
| fx.py | ||