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
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

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

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

3 Commits (e1b38e74dd0072f9d6d48f7c90e21912fe3e85cd)