mirror of https://github.com/llvm/torch-mlir
ccc858f531
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. |
||
---|---|---|
.. | ||
compile_api | ||
eager_mode | ||
torchscript_e2e_test | ||
CMakeLists.txt | ||
annotations-sugar.py | ||
lit.cfg.py | ||
lit.site.cfg.py.in |