torch-mlir

Commit Graph

Author	SHA1	Message	Date
Dave Liddell	04be6ba773	Make the onnx importer more robust for internal/external and large models (#2794 ) Fix for https://github.com/llvm/torch-mlir/issues/2765 The onnx docs say that you can't do shape inference using the in-memory API for models > 2 GB. This fix replaces that API with the file-based API. Since the new API generates an intermediate file, also added a --keep switch to keep that file, which I delete by default. --------- Co-authored-by: Dave Liddell <dliddell@xilinx.com>	2024-01-31 21:58:43 -08:00

Author

SHA1

Message

Date

Dave Liddell

04be6ba773

Make the onnx importer more robust for internal/external and large models (#2794 )

Fix for https://github.com/llvm/torch-mlir/issues/2765

The onnx docs say that you can't do shape inference using the in-memory
API for models > 2 GB. This fix replaces that API with the file-based
API. Since the new API generates an intermediate file, also added a
--keep switch to keep that file, which I delete by default.

---------

Co-authored-by: Dave Liddell <dliddell@xilinx.com>

2024-01-31 21:58:43 -08:00

1 Commits (d5693b3f51a8414cddf4e486daff52d8fa87cfa5)