torch-mlir

Commit Graph

Author	SHA1	Message	Date
Vivek Khandelwal	22c9008bb9	build: Update Roll PyTorch version (#3548 ) This commit also updates the PyTorch and Torchvision nightly links since they are now moved to a different location. PyTorch Nightly: https://download.pytorch.org/whl/nightly/cpu/torch/ Torchvision Nightly: https://download.pytorch.org/whl/nightly/cpu/torchvision/ Disables dtype checks for some ops, tracked by https://github.com/llvm/torch-mlir/issues/3552 Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>	2024-07-19 21:38:57 +05:30
Matthew Francis-Landau	fe9db78120	Allow custom ops to return an array of tensors (#3531 ) This PR adds support to `fx_importer.py` for handling custom ops that return an array of tensors. As long as the length of the array is consistent across runs (determined statically), then this patch will work. This does not require that the number of tensors returned is determined by the op's definition. CC @sjain-stanford	2024-07-14 11:54:23 -07:00
Sambhav Jain	7e0e23c668	Test custom op import with symbolic shapes (#3431 ) Tests the basic constructs of registering a custom op and its abstract implementations (with FakeTensors) in python, going through TorchDynamo export, followed by importing the shape expressions in the Torch dialect. Also fixes the importer were previously the symbolic bind op insertion was not gated in one place.	2024-06-09 00:32:49 -07:00

Author

SHA1

Message

Date

Vivek Khandelwal

22c9008bb9

build: Update Roll PyTorch version (#3548 )

This commit also updates the PyTorch and Torchvision nightly links since
they are now moved to a different location.

PyTorch Nightly: https://download.pytorch.org/whl/nightly/cpu/torch/
Torchvision Nightly:
https://download.pytorch.org/whl/nightly/cpu/torchvision/

Disables dtype checks for some ops, tracked by https://github.com/llvm/torch-mlir/issues/3552

Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>

2024-07-19 21:38:57 +05:30

Matthew Francis-Landau

fe9db78120

Allow custom ops to return an array of tensors (#3531 )

This PR adds support to `fx_importer.py` for handling custom ops that
return an array of tensors. As long as the length of the array is
consistent across runs (determined statically), then this patch will
work. This does not require that the number of tensors returned is
determined by the op's definition.

CC @sjain-stanford

2024-07-14 11:54:23 -07:00

Sambhav Jain

7e0e23c668

Test custom op import with symbolic shapes (#3431 )

Tests the basic constructs of registering a custom op and its abstract
implementations (with FakeTensors) in python, going through TorchDynamo
export, followed by importing the shape expressions in the Torch
dialect.

Also fixes the importer were previously the symbolic bind op insertion
was not gated in one place.

2024-06-09 00:32:49 -07:00

3 Commits (6e8c7bed4b12117764274e79bc60a93443d5bdd5)