llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
1,117 Commits
47 Branches
1195 Tags
79 MiB
Commit Graph

2 Commits (5fbf2a376c3d36f1f31a2cda8a242bd36f090383)

Author SHA1 Message Date
Sean Silva 2af53ce434 torch_mlir.compile: Add OutputType.RAW 
This can help with development and reporting bugs.
 2022-05-19 03:41:43 -07:00
Sean Silva ef9e4c95f2 torch_mlir.compile: add support for dynamic sizes. 
We do this by inroducing a TensorPlaceholder class, which can be used to
specify dynamic sizes. Internally, we canonicalize all example inputs
to TensorPlaceholder's.

This commit also adds some basic testing, which was missing before.
 2022-05-17 07:02:32 -07:00