llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/python/test/compile_api
History
Sean Silva ccc858f531 torch_mlir.compile: Fix API footgun 
use_tracing=True was behaving unexpectedly because the handling of
single arguments was happening after the torch.jit.trace call.

This also fixes the check to specifically test for a torch.Tensor or
TensorPlaceholder so that both lists and tuples would be correctly
handled.
 		2022-06-05 18:10:07 -07:00
..
basic.py torch_mlir.compile: Add OutputType.RAW 2022-05-19 03:41:43 -07:00
tracing.py torch_mlir.compile: Fix API footgun 2022-06-05 18:10:07 -07:00