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> |
||
|---|---|---|
| .. | ||
| sparsity | ||
| v2.3 | ||
| basic_test.py | ||
| custom_op_test.py | ||
| symbolic_shape_expr_test.py | ||
| sympy_to_affine_expr_test.py | ||