torch-mlir

Commit Graph

Author	SHA1	Message	Date
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

Author

SHA1

Message

Date

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

1 Commits (08fc2d89bbda3ca160bc7e7fe73ac97f1d18be94)