torch-mlir
torch-mlir copied to clipboard
llvm
The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
Hi, I have some special self-define torch ops, and want to lower them to torch dialect. Does torch-mlir provide the export mechanism to support self-define torch ops ？ When export...
Both `ElementwiseReluModule` and `ResNet18StaticModule` TOSA tests fail after updating LLVM. See [CI log](https://github.com/llvm/torch-mlir/runs/7849271902?check_suite_focus=true#step:12:680) for details. ``` FAIL - "ElementwiseReluModule_basic" @ trace item #0 - call to "forward" @ output of...
Here's transposed support for convolution, both strided and unstrided.
I'm looking to add support for dynamic shapes with multiple indexing tensors for the aten.index.Tensor op. The semantics for the op are summarized as follows. ``` IndexTensor for multiple input...
Hi I am working on providing the support for [v-diffusion-pytorch](https://github.com/crowsonkb/v-diffusion-pytorch) model via torch-mlir. As of now, I am able to get the simplified version of the model compiled successfully via...
PyTorch recently renamed the shape function `mean_dim` with the name `sum_mean_dim`, and they added the right handling of the optional `dim` argument (https://github.com/pytorch/pytorch/commit/c8eae2de5200b95d59ab090a37eed96528c877ff). This commit updates the shape functions that...
(ignore the "Rework how global slot initializers work." commit -- this one doesn't trivially commute with it, but is still logically separate) This introduces a new pass LowerToBackendContract (better name...
Reverts https://github.com/llvm/torch-mlir/pull/1183 after addressing nightly build failure
To run locally: ``` # Build an image and launch an interactive docker container ./build_tools/docker/run_docker.sh # Run cmake build (either in-tree or out-of-tree) ./build_tools/docker/run_cmake_build{_oot}.sh # Run torch-mlir unit tests (+...
