torch-mlir/python/torch_mlir
Stella Laurenzo 826786bdd0
[fx] Support ExportedProgram buffer mutation. (#3080)
In the prior state when I supported mutation of user inputs by treating
them as mutable-tensor SSA values, I had left the case of buffer
mutation only vaguely implemented until a concrete use emerged.
    
This patch reworks this buffer mutation support by assuming that buffers
must be resolved via the hooks symbolically and treated with load/store
semantics. This is implied in the structure since we have no SSA value
that represents a buffer and we already assume that reading parameters
happens via such a mechanism.
2024-04-01 14:18:12 -07:00
..
_mlir_libs [ci] Upgrade to new runners and disable unsupported jobs. (#2818) 2024-01-27 18:35:45 -08:00
dialects Upstream the ONNX importer. (#2636) 2023-12-12 19:02:51 -08:00
extras [fx] Support ExportedProgram buffer mutation. (#3080) 2024-04-01 14:18:12 -07:00
tools/import_onnx [onnx] Add testing using the `onnx` compilation using torch tests (#2795) 2024-02-15 10:17:13 -08:00
fx.py [fx] Implement auto_functionalized higher order op. (#3063) 2024-03-26 17:06:05 -07:00