torch-mlir

Commit Graph

Author	SHA1	Message	Date
Sean Silva	31727f81d8	torch_mlir.compile: Allow ignoring traced shapes In some cases, users know that a traced graph is valid for a wider set of shapes than they originally traced it with. Provide an option for users to ignore the shapes in the traced graph when they know it is legal. Fixes #997	2022-08-04 10:18:34 -07:00
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

31727f81d8

torch_mlir.compile: Allow ignoring traced shapes

In some cases, users know that a traced graph is valid for a wider set
of shapes than they originally traced it with. Provide an option for
users to ignore the shapes in the traced graph when they know it is
legal.

Fixes #997

2022-08-04 10:18:34 -07:00

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

2 Commits (bb47c166a047ddbc913950cb0f2571594c0fe261)