llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
2,045 Commits
47 Branches
1195 Tags
93 MiB
Commit Graph

2045 Commits (0244f540a787f9219b48877e5d4baddb29066e56)
 

Author SHA1 Message Date
Prateek Gupta 207229297e
[TORCH-MLIR] Add ODS for aten.clamp.Tensor op. (#1894) 
This commit adds the ODS definition for the aten.clamp.Tensor op.

Signed-off-by: Prateek Gupta <prateek.gupta2@cerebras.net>
 2023-02-24 09:18:24 -08:00
Gaurav Shukla 0afb85d45f build: update llvm tag to 5e111eb275 
- Update llvm tag to 5e111eb275eee3bec1123b4b85606328017e5ee5
- mhlo now points to a99159c45ee5c497f8dce01eff807a6d57629b61

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2023-02-23 22:57:17 +05:30
Vivek Khandelwal 6a3438f672 build: manually update PyTorch version 
Set PyTorch and TorchVision version to nightly release 2023-02-20.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 2023-02-23 11:33:41 +05:30
Zachary Cetinic e7111d473b
[Torch Dialect] Scatter reduce lowering (#1884) 
- Lowers the torch.scatter_reduce to linalg_on_tensors dialect.
- Includes support for "sum", "prod", "amax", "amin" and "mean".
 2023-02-21 23:05:55 +00:00
Ramiro Leal-Cavazos 52dbb160fc
Replace `torch.rand` and `torch.randn` in e2e tests with `tu.rand` (#1890) 
Random tensors used in e2e tests should be created using the
`TestUtils` object passed to the registered test case to ensure that
the compiled module and the golden trace receive the same tensors as
input. This commit changes all the cases of `torch.rand` and
`torch.randn` to use the `TestUtils` instead.
 2023-02-21 14:30:05 -08:00
Yuanqiang Liu eb74014dd8
[Torch] decompose aten.norm.ScalarOpt_dim to aten.linalg_vector_norm (#1849) 2023-02-20 20:08:29 -08:00
Ashay Rane 268364e061
CI: install `unzip` before using it (#1893) 
The RollPyTorch action needs the `unzip` command to peek into WHL files
for fetching metadata.  This patch makes sure that the command is
installed before referencing it.
 2023-02-19 17:49:08 -06:00
Ziheng Jiang 38ed559398
[StableHLO] Add support for AtenPowTensorScalar. (#1883) 
* [MHLO] Add support for AtenPowTensorScalar.

* Update.

---------

Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
 2023-02-16 20:26:46 -08:00
Chi_Liu e85def790c
[TOSA] Add todtype f64 and i1 to f32 support (#1845) 2023-02-16 09:10:11 -08:00
powderluv 5710871f4f
Update buildAndTest.yml (#1881) 
* Update buildAndTest.yml

* Update oneshotSnapshotPackage.yml

* Update buildRelease.yml

* Update RollPyTorch.yml

* Update oneshotSnapshotPackage.yml

* Update buildAndTest.yml
 2023-02-15 09:17:12 -08:00
Vivek Khandelwal b17d4d4f08
[MLIR][TORCH] Add decomposition for aten.bernoulli.p op (#1882) 
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 2023-02-15 22:36:29 +05:30
Vivek Khandelwal f6f2e4d040 [MLIR][TORCH] Add support for integer type input for max.dim op 
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 2023-02-15 16:14:15 +05:30
Chi_Liu 8a7340dfb5
[TOSA] aten.index.tensor multiple indexes support (#1868) 2023-02-13 23:07:15 -08:00
Ashay Rane 67ab708b63
python: separate build- and test-related pip dependencies (#1874) 
We want to ensure that pip packages required for building torch-mlir
should be included in the dependencies of torch-mlir, but we don't want
the pip packages required for _testing_ of torch-mlir to be included
among the dependencies.  To be able to specify and install one set of
dependencies and not the other, this patch separates the pip packages
into two files: build-requirements.txt and test-requirements.txt.

This patch also updates references to the requirements.txt file so that
CI builds that run end-to-end tests install test-related pip
dependencies while everything else (including WHL builds) sticks to just
the build-related pip dependencies.

Despite this change, this patch should not affect a torch-mlir
developer's workflow.  More precisely, since this patch makes the
top-level requirements.txt file refer to both build-requirements.txt and
test-requirements.txt files, a torch-mlir developer should be able to
continue referring to the requirements.txt file without any impact.
 2023-02-13 21:22:09 -06:00
powderluv 57b252107e
test torch / torchvision compat (#1877) 2023-02-13 15:26:44 -08:00
Maksim Levental 2eddb3fde7
WIP: No PyTorch dep (#1854) 2023-02-13 14:21:06 -06:00
Roll PyTorch Action 3cf5f4fcb7 update PyTorch version to 2.0.0.dev20230213 2023-02-13 15:23:12 +00:00
Roll PyTorch Action be4d220e81 update PyTorch version to 2.0.0.dev20230212 2023-02-12 14:45:19 +00:00
Roll PyTorch Action bb8fbc41e7 update PyTorch version to 2.0.0.dev20230211 2023-02-11 14:40:15 +00:00
Yuanqiang Liu 6ab990e1e8
[Torch Dialect] add folder for aten.Int.float (#1863) 2023-02-10 13:59:03 -08:00
Ziheng Jiang f1b8d5e581
[MHLO] Support AtenMaskedFillScalar (#1839) 
* [MHLO] Support MaskedFillScalar.

* Update.

* Update.

* Update.

---------

Co-authored-by: Ziheng Jiang <ziheng.jiang@bytedance.com>
 2023-02-10 13:58:39 -08:00
Yuanqiang Liu 2f6fdb7f0b
[Torch Dialect] add folder for prim.min.int (#1864) 2023-02-10 13:58:15 -08:00
powderluv 320e67ff34
Python 3.11 support (#1848) 
* Python 3.11 support

* test without torchvision

* Update pytorch-requirements.txt

* Update buildRelease.yml

* Update action.yml

* Update install_macos_deps.sh

* Update build_macos_packages.sh
 2023-02-10 07:16:37 -08:00
Roll PyTorch Action a7294432bb update PyTorch version to 2.0.0.dev20230210 2023-02-10 14:34:25 +00:00
Kunwar Grover 230fc3ffb6
Add compile_commands.json to gitignore (#1867) 2023-02-10 06:26:49 -08:00
Chi_Liu cc819e73dd
[TOSA] Fix broadcast_to input and output different shape support (#1855) 2023-02-09 09:15:14 -08:00
Roll PyTorch Action 83534370c3 update PyTorch version to 2.0.0.dev20230209 2023-02-09 15:23:33 +00:00
Roll PyTorch Action eb7bac7ca4 update PyTorch version to 2.0.0.dev20230208 2023-02-08 16:14:29 +00:00
Vivek Khandelwal 282c4e027b build: manually update PyTorch version 
Set PyTorch and TorchVision version to nightly release 2023-02-07.
Fixes https://github.com/llvm/torch-mlir/issues/1792.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 2023-02-08 11:41:58 +05:30
Zachary Cetinic 2a4a61f98f
Add aten.scatter_reduce op definition (#1846) 2023-02-07 21:59:07 +00:00
Tanyo Kwok 3ebe5a5a67
build: update llvm tag to ba8b8a73f (#1856) 
* build: update llvm tag to ba8b8a73f

* skip ConvolutionModule2DGroups_basic
 2023-02-07 23:25:59 +08:00
Vivek Khandelwal 3e60e6021f build: manually update PyTorch version 
Set PyTorch and TorchVision version to nightly release 2023-02-06.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 2023-02-07 12:02:18 +05:30
Yuanqiang Liu 089018b658
[MHLO] move AtenTanhOp to ConvertAtenUnaryFPOnlyPatten and add sin/cos/ceil/floor pattern (#1847) 2023-02-06 11:14:26 -08:00
Vivek Khandelwal c957cebd03 build: manually update PyTorch version 
Set PyTorch and TorchVision version to nightly release 2023-02-05.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 2023-02-06 13:23:28 +05:30
Zachary Cetinic 2c2009a13d
Add in-place variant of torch.scatter_add (#1836) 2023-02-03 17:54:28 +00:00
Jiahao Li f58ba19448
Add aten.bucketize op and its decomposition (#1834) 2023-02-03 10:20:47 +08:00
Roll PyTorch Action 5d55390111 update PyTorch version to 2.0.0.dev20230202 2023-02-02 15:06:52 +00:00
Ashay Rane 711646d095
mhlo: migrate conversion to stablehlo (#1840) 
This patch replaces all MHLO operations with their StableHLO
counterparts and adds a validation pass to ensure that no MHLO operations
remain before translating all Stablehlo operations to the MHLO dialect
for further lowering to the Linalg dialect.

This patch also updates all lit tests so that they refer to the
`convert-torch-to-stablehlo` pass and so that they check for StableHLO
operations.
 2023-02-02 07:29:47 -06:00
Vivek Khandelwal ed9d8d1fb7 [MLIR][TORCH] Add support for clone op with channels last memory format 
Fixes https://github.com/llvm/torch-mlir/issues/1829

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
 2023-02-02 16:04:42 +05:30
Ashay Rane 95361747c2
Revert "update PyTorch version to 2.0.0.dev20230201" (#1842) 
This reverts commit ca400e02ce, since the
(upstream) Windows build of torchvision has an incorrect dependency.
 2023-02-01 13:43:53 -06:00
Roll PyTorch Action ca400e02ce update PyTorch version to 2.0.0.dev20230201 2023-02-01 16:27:37 +00:00
Roll PyTorch Action 5057e88715 update PyTorch version to 2.0.0.dev20230131 2023-01-31 14:40:23 +00:00
Sean Silva 72fbf316b4 Update LLVM and MHLO submodules. 
Week of 01/30/2023:

Green LLVM commit: e31ee6417c33a6e2f0e8440b1a86d5365279ad68
Green MHLO commit: c2a6f4064d426567b9ef7b0d29d5ab86dc7b2b02 (branch greencommit/2023-01-30-e31ee641)
 2023-01-31 06:08:21 -08:00
Roll PyTorch Action c622f59300 update PyTorch version to 2.0.0.dev20230130 2023-01-30 14:33:24 +00:00
Jiahao Li f5b689e12f
[MHLO] Support aten.cumsum op in mhlo backend (#1825) 2023-01-29 21:38:27 -08:00
Roll PyTorch Action 6040d7ce00 update PyTorch version to 2.0.0.dev20230129 2023-01-29 14:32:31 +00:00
Roll PyTorch Action ef4316e046 update PyTorch version to 2.0.0.dev20230128 2023-01-28 14:34:24 +00:00
Chi_Liu 00fc14a6e1
[TOSA] Add to.dtype i1 to i64 (#1830) 2023-01-27 09:21:06 -08:00
Roll PyTorch Action 31ffb815e4 update PyTorch version to 2.0.0.dev20230127 2023-01-27 14:25:52 +00:00
Roll PyTorch Action 37a625e737 update PyTorch version to 2.0.0.dev20230126 2023-01-26 14:56:31 +00:00