Ecosyste.ms: OpenCollective

An open API service for software projects hosted on Open Collective.

github.com/llvm/torch-mlir

The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
https://github.com/llvm/torch-mlir

[LINALG] Lower `aten.Matmul` to `linalg.BatchMatmul`

This commit lowers `aten.matmul` to `linalg.BatchMatmul` under the
following conditions:
1. The ...

1be604bfd3ce3699598bacb844f115701414d453 authored over 2 years ago by Gaurav Shukla <[email protected]>
development.md: updating the LLVM submodule

We recently noticed that it is necessary to update the shape lib while
doing an LLVM bump: https...

f774e63abdcbe786e884d0cc14ba198192387332 authored over 2 years ago by Sean Silva <[email protected]>
Include Bazel BUILD steps to development.md (#954)

* Add bazel build steps

* move to development.md

* reorder

3bb2d548a5171d9c32b134f63945834596c69691 authored over 2 years ago by Sambhav Jain <[email protected]>
[LINALG] Fix shape function of index.Tensor + support N-rank inputs (#972)

This commit fixes the shape function for `index.Tensor`, adding
support for multiple index tens...

400fecc1e55c15ad4688c9222e8307b9a0eab1bf authored over 2 years ago by Ramiro Leal-Cavazos <[email protected]>
linalg: lower `aten.triu` op to `linalg.generic` (#965)

Prior to this patch, the torch dialect included `AtenTriuOp` for
computing the upper triangular...

234fc7fe0c402367d4917787402e7c6afd16b8f6 authored over 2 years ago by Ashay Rane <[email protected]>
[MLIR][TORCH] Add folder for torch_c.from_i64 & torch_c.to_i64 (#933)

* [MLIR][TORCH] Add folder for torch_c.from_i64 & torch_c.to_i64

* add unit tests for each in...

143a7bcb7632d1bdb7a3cc62332a5eb2f647be70 authored over 2 years ago by Tanyo Kwok <[email protected]>
Update shape library with LLVM bump changes (#973)

189afa82c5d0284bbf7a45f8cec75a1cda9659c3 authored over 2 years ago by Ramiro Leal-Cavazos <[email protected]>
Bump LLVM (#970)

Adds my grouped conv2d op for #858

38650812fdfe47bdac6017bcdc1fbefbe1f36dac authored over 2 years ago by gpetters94 <[email protected]>
Add canonicalization for aten.add.tensor op

5cff40c88a1468752a23a7c9971cfa6a5c045351 authored over 2 years ago by erman-gurses <[email protected]>
Bump LLVM (#966)

e143a34948dcc845cddd8d21f5f13862520b1e0e authored over 2 years ago by Maksim Levental <[email protected]>
Bump LLVM (#958)

829717c96eb09e6393813950532c71c2789718cc authored over 2 years ago by Maksim Levental <[email protected]>
[LINALG] Fix handling of size-1 dims in `aten.view` (#962)

This commit adds support for several size-1 dims in a row in an
expansion using `aten.view`.

8b9475930399fb70f2cd929a26a47650b6ce0542 authored over 2 years ago by Ramiro Leal-Cavazos <[email protected]>
Fix `verifyLinalgCompatibleTypes` which currently doesn't successfully catch `torch.tensor`. (#947)

a34dad2e077592deb497a9077fc3188b6e1154d5 authored over 2 years ago by Maksim Levental <[email protected]>
[MLIR][TORCH] Add decomposition of aten.numpy_T op

This commit adds the decomposition of `aten.numpy_T` op into
`aten.t` or `aten.permute` op.

Sig...

77ab31641fde8cabd6403648080c69d315ffd67b authored over 2 years ago by Vivek Khandelwal <[email protected]>
[MLIR][TORCH] Add support for bool type in convertScalarToDtype utility

Signed-Off By: Vivek Khandelwal <[email protected]>

4605dc9c99bf56d00e335f1e8e703347c75bf541 authored over 2 years ago by Vivek Khandelwal <[email protected]>
Add E2E support for aten.is_floating_point

708a51ae2e70d3d4756b89200e26f42a6c753177 authored over 2 years ago by Albert Sandru <[email protected]>
[LINALG] Fix typo in conversion pattern of `aten.embedding` (#942)

246c2df65a01998b2159fec79675a0631f1f0218 authored over 2 years ago by Ramiro Leal-Cavazos <[email protected]>
Update shape lib docs.

2a0c5de3633ec2df1c939eac15fbafcd64a3c968 authored over 2 years ago by Sean Silva <[email protected]>
[MLIR][TORCH] Add integer dtype support for aten.rsub.Scalar op

Signed-Off By: Vivek Khandelwal <[email protected]>

aed5517fda2551a68e6e606dd1acf2dd3d507a8c authored over 2 years ago by Vivek Khandelwal <[email protected]>
Temporarily revert support for custom op extensions. (#944)

The MacOS builders are having linking trouble with the extension library.
Until it's fixed, all...

b90837ee243b086bd487119da4e140b2df397498 authored over 2 years ago by Bob Adolf <[email protected]>
Update CMakeLists.txt

8fd084377dbd9d4b17681256c75db019db81af51 authored over 2 years ago by powderluv <[email protected]>
Update CMakeLists.txt

Emergency fix to unblock the nightly Release builder

dfc6f7c547fcaa5c56a1e007648fcb7873f16885 authored over 2 years ago by powderluv <[email protected]>
[LINALG] Add 0-rank case for `aten.permute` (#940)

The function `AffineMap::inferFromExprList` does not work if the first
vector of expressions is...

93f6d8e77679016dab45e77ee2500509d79cc95c authored over 2 years ago by Ramiro Leal-Cavazos <[email protected]>
[MLIR][TORCH] Add decomposition of aten.floor_divide op

This commit adds the decomposition of `aten.floor_divide` op into
`aten.div.Tensor_mode` op.

Si...

33fa8e7761321bcedf4c96b65ebba655fb1d1555 authored over 2 years ago by Vivek Khandelwal <[email protected]>
Fix: 0 sizes tensor being regarded as unknown rank (#923)

0d4445eaf9e270207b3670aa3eb4316e489c59d4 authored over 2 years ago by Tanyo Kwok <[email protected]>
Allow torch-mlir to support PyTorch extensions. (#895)

PyTorch allows new operators to be registered dynamically in modules.
Torch-mlir already makes ...

0a7ba62438275b02a42ee30b04e2152a51f26ff7 authored over 2 years ago by Bob Adolf <[email protected]>
Update README.md

4bf34523d779f62276c8f0e4250e4f7ef18b6746 authored over 2 years ago by powderluv <[email protected]>
Revert requirements.txt (#930)

https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html

is going to be deprecated via...

1adc0f166125aae7ae7fce93616df4e9e2f454e0 authored over 2 years ago by powderluv <[email protected]>
Change to the real PackedParams.h location (#929)

Also update the PyTorch nightly URL

02b917f769b3fc063abecb7523561149d7e7246e authored over 2 years ago by powderluv <[email protected]>
Fix new location for PackedParams.h (#928)

Looks like they renamed it in location

4cdf4e7d47aed8fafacb764f4806c02305d39dbe authored over 2 years ago by powderluv <[email protected]>
Fix class_annotator_pybind.h header guard (#924)

merging to unblock builders

e70d4f732d713567904bb6f27cb7ac3a9471bd69 authored over 2 years ago by Tanyo Kwok <[email protected]>
Fix the new header location (#926)

Seems to have moved in the latest nightly

6615add806eaafcfe61478049e3b3998290abbda authored over 2 years ago by powderluv <[email protected]>
Handle `nn.Linear(..., bias=False)` case for TorchToLinalg (#919)

5c85ac3100338e32309fefda87b4366de2dc1a41 authored over 2 years ago by Maksim Levental <[email protected]>
Use double quotes instead of single quotes (#918)

298d095acf6c6cd9e14d4af68522bbb35a89f581 authored over 2 years ago by Henry Tu <[email protected]>
Generate underscore variant of functional ops (#915)

* Generate underscore variant of functional ops

* Do not apply `IsTrailingUnderscoreInplaceVa...

c1da9edcf044a58cddbde3f982d216a4a8b95192 authored over 2 years ago by Henry Tu <[email protected]>
Remove pybind deps from importer and annotator (#903)

* Remove pybind deps from importer and annotator
* Rename files to class_annotator_pybind.cpp/.h

bd53998da8d3be5371845f3ffbbf093b38c0b6db authored over 2 years ago by Tanyo Kwok <[email protected]>
Use upstream shape functions directly.

Now that upstream exposes them nicely, we can use them.

I noticed that we had added stuff into ...

e1b38e74dd0072f9d6d48f7c90e21912fe3e85cd authored over 2 years ago by Sean Silva <[email protected]>
Update debug options in compilation errors (#913)

The flag for printing the IR after each pass is now prefixed with
"mlir". This commit updates t...

22c0893ec6220c89d1ce63b276afef085e232fc6 authored over 2 years ago by Ramiro Leal-Cavazos <[email protected]>
[MLIR][TORCH] Add E2E support for aten.div.Tensor_mode op

This commit adds lowering of `aten.div.Tensor_mode` op.
This commit also fixes formatting for th...

b95b3d844d1ea6710f316ab7d140adf83f67f526 authored over 2 years ago by Vivek Khandelwal <[email protected]>
[MLIR][TORCH] Add E2E support for aten.baddbmm op

This commit decomposes `aten.baddbmm` op into `aten.bmm`,
`aten.mul.Scalar`, and `aten.add.Tenso...

a11ef674a7373c3574752a37b4853f3fccecce2f authored over 2 years ago by Vivek Khandelwal <[email protected]>
Add Support for aten::scatter_add (#906)

fe784fd90049b2b2f3d1c51a8d712a34d51b3b42 authored over 2 years ago by Jae Hoon (Antonio) Kim <[email protected]>
Add support for aten::arange.start_out (#905)

8a1839a17e1c52080da168bece005a4be3a1c3c1 authored over 2 years ago by Jae Hoon (Antonio) Kim <[email protected]>
[MLIR][TORCH] Add E2E support for aten.clamp_[min|max] op

This commit decomposes `aten.clamp_min` and `aten.clamp_max` op
into `aten.clamp` op.

Signed-Of...

2718b4d838c2f9aabebc3828c6341b949fd677ae authored over 2 years ago by Vivek Khandelwal <[email protected]>
torch_mlir.compile: Fix API footgun

use_tracing=True was behaving unexpectedly because the handling of
single arguments was happenin...

ccc858f5311fdcdb7828d691a9db3bd565ac52f5 authored over 2 years ago by Sean Silva <[email protected]>
bump macOS builds to Python 3.10

2f0b1d0b08f0fa296b58563af4a03a1ab8e122bc authored over 2 years ago by powderluv <[email protected]>
Build the nightly package only once a day/night

No need to be shipping two releases a day, our supported packages and binaries have grown.

b14c5d619daa5cff8d328985eb338dbd93826eb0 authored over 2 years ago by powderluv <[email protected]>
Add E2E support for AtenLogicalOrOp. (#883)

fc419b1e7d966b9589b021f1c6f02136482c5c3c authored over 2 years ago by Vidush Singhal <[email protected]>
Replace valsem.aten.zero with aten.zero.functional (#893)

abf5c94a1bf80cbdad4414d68f6aa59bd70d80e8 authored over 2 years ago by Henry Tu <[email protected]>
Added support for native_dropout_backward (#892)

650f5a5008e3082edd05d7a01cb58a8671d7c4fc authored over 2 years ago by Henry Tu <[email protected]>
Added support for native_dropout (#891)

b7082a8d4ec1168f05271b6f16f592932e14c640 authored over 2 years ago by Henry Tu <[email protected]>
Added support for native_batch_norm_backward (#890)

a635fd228775e676c8cc113ec9fad6412b77bfb9 authored over 2 years ago by Henry Tu <[email protected]>
Added support for embedding_dense_backward (#889)

bfe8ff4b426d22261c53f599ea7ccd1a28802525 authored over 2 years ago by Henry Tu <[email protected]>
Added support for native_layer_norm_backward (#888)

a29903dfc8b55cc206b01790a850fb4bc7c5e9e8 authored over 2 years ago by Henry Tu <[email protected]>
Add E2E support for AtenAllBoolOp (#874)

0a913bc904c2752c4357dcc1670d315339bfae8b authored over 2 years ago by Vidush Singhal <[email protected]>
build: remove manual changes to ShapeLibrary.cpp (#894)

The patch bumped up the LLVM tag made manual fixes to the code in
`ShapeLibrary.cpp`. However,...

7fdc1cff02ecd2e645be6fb5cd3de2dbfb417994 authored over 2 years ago by Ashay Rane <[email protected]>
[MLIR][TORCH] Add decomposition of aten.adaptive_avg_pool2d op

This commit adds the decomposition of `aten.adaptive_avg_pool2d` op into
`aten.avg_pool2d` op. T...

6f548fc3ada3c7cb45240c3ac90d9529b997a1eb authored over 2 years ago by Vivek Khandelwal <[email protected]>
[tosa] Support for AtenAvgPool2d op

Signed-Off By: Vivek Khandelwal <[email protected]>

06750815d14371a7b8f0e964f2380517583e745f authored over 2 years ago by Vivek Khandelwal <[email protected]>
Emit `aten.unsqueeze` with mutating variants (#873)

The op `aten.unsqueeze` has a mutating variant. This commit adds
support for that variant.

b76c8c82dc9aaaeeb9f2e26e339c6879a9672133 authored over 2 years ago by Ramiro Leal-Cavazos <[email protected]>
build: fix code so that the compiler does not emit warnings (#871)

When compiling without assertions (i.e. in `NDEBUG` mode), a handful of
statements turn to NOPs...

029cd5432752a3d53acb459fd57ea9158cfa7a73 authored over 2 years ago by Ashay Rane <[email protected]>
add ci tests (#754)

cec5aeedb0b11fd5d49683286d0800c61ad5b795 authored over 2 years ago by Maksim Levental <[email protected]>
Update development.md

24e04d5729d251c9f0cf0df85308de43a7b646a3 authored over 2 years ago by powderluv <[email protected]>
[MLIR][TORCH] Add E2E support for aten.Bool.[float|int] op

This commit adds lowering of `aten.Bool.float` and `aten.Bool.int` op.

Signed-Off By: Vivek Kha...

56e77d4213b6bca8c6e365bac6ecbc329d2c24d9 authored over 2 years ago by Vivek Khandelwal <[email protected]>
[MLIR][TORCH] Add E2E support for aten.any.bool op

This commit adds lowering of `aten.any.bool` op.

Signed-Off By: Vivek Khandelwal <vivek@nod-lab...

014a6d16c7891d194e9031acba21b85e2a89453e authored over 2 years ago by Vivek Khandelwal <[email protected]>
[MLIR][TORCH] Add E2E support for aten.sqrt.int op

This commit adds lowering of `aten.sqrt.int` op.

Signed-Off By: Vivek Khandelwal <vivek@nod-lab...

bc9b2156e33c194dd6e94547feda2c1cf7003927 authored over 2 years ago by Vivek Khandelwal <[email protected]>
development.md: Add details about testing.

There is more detail to be added here, but this is a start.

f791b2ecaeea759f4f17aa140ebb113078c061cd authored over 2 years ago by Sean Silva <[email protected]>
Add weekly meetings to the readme.

30adb1e6750d3f814f91ef90deb76ec0ac30cbdf authored over 2 years ago by Sean Silva <[email protected]>
torch,linalg: add support for translating aten.linalg.vector_norm (#839)

This patch adds support for the torch.linalg.vector_norm op to the torch
dialect, including the...

f18b2be9111aa1c199d42ab61e1e1c8d06bf5a02 authored over 2 years ago by Ashay Rane <[email protected]>
torch.prim.TupleIndex: Adjust tensor types when folding.

In cases where a refinement/derefinement was needed, we didn't fold.

Fixes https://github.com/l...

3fb54cba4cb1284b744dd55a2eb38d3a9f3b9975 authored over 2 years ago by Sean Silva <[email protected]>
torch_mlir.compile: Add OutputType.RAW

This can help with development and reporting bugs.

2af53ce43437dd3cf00dc592f62e9654cd53ba9e authored over 2 years ago by Sean Silva <[email protected]>
Add simple neural_net and bert_training scripts.

1. With the help of `make_fx` we are able to get the full training graph
with weight updates....

10c8e3c5936c43eff31bfe929eeffa2996494ea9 authored over 2 years ago by Prashant Kumar <[email protected]>
torch_mlir.compile: add support for dynamic sizes.

We do this by inroducing a TensorPlaceholder class, which can be used to
specify dynamic sizes. ...

ef9e4c95f2f4cafa6903b61b1d8246c01234ea99 authored over 2 years ago by Sean Silva <[email protected]>
mlir: bump llvm tag to 5380e3 (#856)

In addition to updating the llvm-project submodule, this patch also:

1. updates shape functio...

bb52a460cb101ed63d73357953fbbb544cd62c59 authored over 2 years ago by Ashay Rane <[email protected]>
build only Python3.9 to avoid timeout

GH runner times out when building 3.9 and 3.10 on macOS.

cfc1a6515c00badbf15bff8e3f2e796d3199b8e0 authored over 2 years ago by powderluv <[email protected]>
Build each OSX python version in an venv (#852)

Previously only system default versions were built. Now we build
binaries for both 3.9 and 3.10

d872f3e2caafda66d961ab09c74a725979358e98 authored over 2 years ago by powderluv <[email protected]>
Add shape info to `rand_like` + support for `dtype` flag (#851)

The op `aten.rand_like` was missing a shape function, unit tests, and
the `dtype` argument was ...

96f90efd1606cc6695d5fb514ee23817bb733174 authored over 2 years ago by Ramiro Leal-Cavazos <[email protected]>
Use delocate to make portable wheels on OSX (#850)

Fix up wheels per python version on OSX

e7f306ec2fc532a43958e4381b3f6f8212a908ea authored over 2 years ago by powderluv <[email protected]>
Add -s flag to run e2e tests sequentially

A user might want to avoid the extra layer of multiprocessing libary for
debugging purpose. In s...

ec0e9e0bc7d1c645c54dbaaf4b7ea09655cb5781 authored over 2 years ago by Yi Zhang <[email protected]>
[MLIR][TORCH] Add support for ceil_mode = true for pooling ops

This commit adds support for aten.max_pool2d, aten.max_pool2d_with_indices,
and aten.avg_pool2d ...

f15d257aac601108193c9ab704d8aeb0238274fc authored over 2 years ago by Vivek Khandelwal <[email protected]>
[MLIR][TORCH] Add E2E support for ScalarImplicit, Int.Scalar op

This commit adds lowering of `aten.ScalarImplicit` and `aten.Int.Scalar` op.

Signed-Off By: Viv...

c69a1e56881272cae312dc55e49f0f7ff94c4fe7 authored over 2 years ago by Vivek Khandelwal <[email protected]>
[TORCH] Add folding of aten.detach op.

`aten.detach` op is folded and returns the first operand since it's an
identity function(kind of...

12b3af70d398901cbf22984450f8337c0b139e20 authored over 2 years ago by Prashant Kumar <[email protected]>
[LINALG] Add support for preserve memory format in aten_empty_like op.

The preserve memory specifies that `If any of the input tensors is in channels_last format,
oper...

2b1b0f6e1970c9db13caea2515070c61d4dee167 authored over 2 years ago by Prashant Kumar <[email protected]>
Workaround to make CI pass

5a6210b35bee3c9e2426c79dd8f2b38ef6d2875d authored over 2 years ago by Yi Zhang <[email protected]>
Update torch_mlir_tensor.py

typo

2e6a9c084ef694311ab331018b06d44e2ff438ad authored over 2 years ago by yuhao <[email protected]>
Fix type promotion code for scalar only operations

Fix the type promotion code for scalar only operation to return
TorchType which is the type trac...

28be6511d253c216e5243b906a1bb777f693e5e6 authored over 2 years ago by Yi Zhang <[email protected]>
[MLIR][TORCH] Modify aten::dropout op description

Signed-Off By: Vivek Khandelwal [email protected]

b20679e1b80524d56b217e5feb79cd81f444f24e authored over 2 years ago by Vivek Khandelwal <[email protected]>
Make e2e testing parallel

This change makes the e2e testing parallel using the multiprocessing
python module.

2ed90741ebe6b8f7dd6af376a6852eb81f7ddddc authored over 2 years ago by Yi Zhang <[email protected]>
[MLIR][TORCH] E2E support for [ge|ceil].float, [ge|ne|gt].float_int op

This commit adds lowering of `aten.ge.float`, `aten.ge.float_int`,
`aten.ne.float_int`, `aten.gt...

96fabc0036fadc521b7d455eb35aaf11e14a93f2 authored over 2 years ago by Vivek Khandelwal <[email protected]>
changed name option to decompose-complex-ops

e682b1d0f3a693f4e2214d4a07cfad79631710a5 authored over 2 years ago by Kristof Denolf <[email protected]>
add no decompose option

5243638e3366ed0eba196f6382c0056ca70d94c7 authored over 2 years ago by Kristof Denolf <[email protected]>
Add support for scalar type propagation

The main changes are:
- Added `ValueKnowledge.scalarType` to track scalar type information.
- Ad...

9f7264a7a4e346fd70b5d2e896b3d9f0102c8e16 authored over 2 years ago by Yi Zhang <[email protected]>
Update build_macos_packages.sh

Set default OSX SDK to 11.0 not 11:0

0fb7a03ac9cf13b83014cecc78e8cc61b0397a95 authored over 2 years ago by powderluv <[email protected]>
[LINALG] Add E2E support for `aten.mean.dim` op

- This commit adds support for `aten.mean.dim` op.
- It also adds a new test script `stats.py` f...

4b911ada403eb2c261c4472a951ada75a47b1c49 authored over 2 years ago by Gaurav Shukla <[email protected]>
Fix TupleIndex canonicalizer.

It would change the result type.

32159c4e545359d4ad9e3b492c07e07528f5b47e authored over 2 years ago by Sean Silva <[email protected]>
Add tracing suport to `torch_mlir.compile`.

This also has a fix for the adjustment of types of TupleConstruct
inputs, which I found when usi...

ab5ad7af09ac1fade2eb27dd3ed5bd1a744e0c67 authored over 2 years ago by Sean Silva <[email protected]>
Mention `python-dev` package in development.md

641b0b3be2d14f5b76bd87aa08b7c083d9b926e9 authored over 2 years ago by Sean Silva <[email protected]>
[MLIR][TORCH] Add E2E support for aten.to.dtype_layout op

This commit decomposes `aten.to.dtype_layout` op into `aten.to.dtype` op.
This commit also fixes...

c0634bc996b983f1bcf27510114c2dc72e124ac8 authored over 2 years ago by Vivek Khandelwal <[email protected]>
Add aten.flip (#817)

c4dcdd1e34d5cf38aa884afd95ffbc2c6000efa5 authored over 2 years ago by gpetters94 <[email protected]>
[MLIR][TORCH] Add E2E support for aten.masked_fill.Scalar op

This commit adds lowering of `aten.masked_fill.Scalar` op.
This commit also fixes the formatting...

8a06419980cecda344f092b0ecb1a594ba437d40 authored over 2 years ago by Vivek Khandelwal <[email protected]>
Remove mention of python-dev.

It belongs in the development docs.

07d92bcbb4439fa5eb524090f4a8a6bd121ac473 authored over 2 years ago by Sean Silva <[email protected]>
Provide a way to override MacOS target and arch (#818)

Useful when we are only building for one architecture.

fe1237b2a414d5b27a013872e7888921dbc57d71 authored over 2 years ago by powderluv <[email protected]>