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

Remove some older docs.

a0ccfcf581158d4a4aec2fd62a23015221991b05 authored about 3 years ago by Sean Silva <[email protected]>
Remove some more old stray files.

3dc9b4ee2f399427991f049dfc6c56198434083c authored about 3 years ago by Sean Silva <[email protected]>
Remove docker stuff.

It's not used in CI and doesn't seem to be up to date.

12dc2fa93310f290108f9edd07e29d1f50b7e9d4 authored about 3 years ago by Sean Silva <[email protected]>
Eliminate almost all mentions of IREE.

A few remain in examples/docs that will be naturally be updated in due
time.

This regresses the...

1a0b953ea7edfaf26fd3881d102c64fe0c1aab29 authored about 3 years ago by Sean Silva <[email protected]>
Remove "refjit" terminology.

We now use RefBackend/refbackend consistently.

8779d920b24ea3a53304998094ff39dcef8fe5b9 authored about 3 years ago by Sean Silva <[email protected]>
Remove old RefBackend

It is superceded by the new one.

a25163fbfa1ef2769b688a522b47d8185e5d8beb authored about 3 years ago by Sean Silva <[email protected]>
Bring up new RefBackend.

`tools/torchscript_e2e_test.sh` is all green.

This needs a few passes I put into torch-mlir/lib...

f9c48d0b89282da2007aa12a779926c9e4025689 authored about 3 years ago by Sean Silva <[email protected]>
WIP implementation of torchfx (#304)

Implements a python package for taking a `torch.fx.GraphModule`
and turning it into MLIR in the...

1f00f95d2e943355bf758b6cab3481ffba79c2d9 authored about 3 years ago by Ramiro Leal-Cavazos <[email protected]>
Update update_torch_ods.sh to use new torch_mlir python layout

19feb58ef876bb854321f5100a4cda0b0defee54 authored about 3 years ago by Ramiro Leal-Cavazos <[email protected]>
Implement Python relayout from #311

Fixes https://github.com/llvm/mlir-npcomp/issues/311

The key change is that TorchPlugin is fold...

6d8e7f1bb1dc7b7b798584f15aff992794d36b87 authored about 3 years ago by Sean Silva <[email protected]>
Added transpose lowering

ecc334123ca745497f422952f0041abbca1b7ec4 authored about 3 years ago by George Petterson <[email protected]>
Build fix: Include the builtin and std dialects in the torch_mlir python build. (#310)

Should fix issue noted in #304

c24ca5d6398f246a82c25e8cd4ca461a58b29182 authored about 3 years ago by Stella Laurenzo <[email protected]>
Remove TypeMapper

3e3459690c36360d59381f758d6b9f496cd56b40 authored about 3 years ago by Sean Silva <[email protected]>
Remove op_builder.h/cpp

6f710bbc47af7cec6cc87246faa17cca09981154 authored about 3 years ago by Sean Silva <[email protected]>
Remove acap_dispatch.

This is old code that barely worked, and this approach just won't scale.
TorchFX seems like the ...

900f0e04aad12c34118b513daf34cba853c1f5ae authored about 3 years ago by Sean Silva <[email protected]>
Factor out definition of the _torch_mlir.get_registered_ops function.

It didn't make sense in the main registration file.

9e2442d6b01150d56cd7abca0296bb9243fd38ac authored about 3 years ago by Sean Silva <[email protected]>
Fix file comment filename

8d27c41f21e5f0cbd8129efd742f7b823d70cb10 authored about 3 years ago by Sean Silva <[email protected]>
Move files out of TorchPlugin/csrc/builder/ directory.

It was an unnecessary layer of indirection -- there was nothing outside
of it, and it was just h...

b738db34cd64c49f9e1dc5d3e3f0155185eb1748 authored about 3 years ago by Sean Silva <[email protected]>
Fold torch_mlir_dialects python package into `torch_mlir`.

After this change, there are now just two subdirectories in the
`python_packages` directory in o...

5f3b1ce0b8dd0ca8933350b6abbed4391c4a4fe8 authored about 3 years ago by Sean Silva <[email protected]>
Remove NPCOMP_ENABLE_IREE CMake flag.

Our new dependency management solution relies:
- on the C++ side with the public iree-dialects p...

68fefe7e1f1f23f877aaefec0809ee02fb868e54 authored about 3 years ago by Sean Silva <[email protected]>
Remove frontends/pytorch directory.

It just contained the e2e testing framework. We now fold it into the
main project to reduce comp...

0eb767ea451d54af0161349a123b354071ff44c4 authored about 3 years ago by Sean Silva <[email protected]>
Again try to fix nondeterminism in check-npcomp-all.

2e63f4b1e11dfac6b43fbf3fcc87e91a944f7c37 authored about 3 years ago by Sean Silva <[email protected]>
Try to fix nondeterminism in check-npcomp-all.

44d615ac1ec57c213f0feed3083cd91a9d2f40b3 authored about 3 years ago by Sean Silva <[email protected]>
Bring CI back to life.

This brings back `check-npcomp-all` and the refbackend e2e tests
coverage.

d94d6800fa0af958d6fdb1bec627924e3693bfc8 authored about 3 years ago by Sean Silva <[email protected]>
[torch-mlir earthmoving (2/N)] Python code movement.

This moves the bulk of the Python code (including the Torch interop)
from `frontends/pytorch` in...

b6be96d7225146bb88f5b278ebb13fb3a59bb416 authored about 3 years ago by Sean Silva <[email protected]>
[torch-mlir earthmoving (1/N)] C/C++ code movement.

This creates the `external/torch-mlir` directory as an
LLVM_EXTERNAL_PROJECTS-compatible project...

28a7738189b9d66c72fc530d27852002d0b32420 authored about 3 years ago by Sean Silva <[email protected]>
Comment out the full wheel build

Last commit was only the last step of that.

28762699b3d797f43e1533450122db7993cf70d5 authored about 3 years ago by Sean Silva <[email protected]>
Temporarily disable wheel building

It will be re-enabled after the torch-mlir excision is completed.

0d8af195508b6dcc1ffb677856e4f16f7564db32 authored about 3 years ago by Sean Silva <[email protected]>
Add basic support for lists.

This plumbs through a vertical slice of support for lists.

The main chunk of new code here is A...

a7252f9a06c98de1508f0c26e6ffc15ea514270f authored about 3 years ago by Sean Silva <[email protected]>
MT model compilation minor changes

This contains the following changes:
- Fix optional knowledge propagation. The initial knowledg...

73d553e16827c16a466ebe18c3a273e55d74a1bc authored about 3 years ago by Yi Zhang <[email protected]>
Fix lowering of reduce ops

We were not filling the `outs` with the neutral element of the
reduction, which resulted in read...

5f3eb637c40f24f86a85d90e6fd031c4c788cb39 authored about 3 years ago by Sean Silva <[email protected]>
Added sum lowering

Added lowering to torch.sum into linalg

6724de7692284f63ea15a1ea86018e93adb06b15 authored about 3 years ago by Ramiro Leal-Cavazos <[email protected]>
Fix import in jupyter notebook.

600cc6b9c7c35c372cbd396eb1976f0083c22aa6 authored about 3 years ago by Sean Silva <[email protected]>
Fix TorchToIREE lowering.

We needed to resize the list, not just reserve capacity.

ed2afe43e757807e5605a0fc6174e9b043fb52b5 authored about 3 years ago by Sean Silva <[email protected]>
Clean up stale examples.

They were confusing users, and most didn't even work anymore.

7a3570e88110151854b8d88fb7cccf2f848606ba authored about 3 years ago by Sean Silva <[email protected]>
Update llvm-project to 830c0b9023cd0cf91955900e0d96283e7a8c3711

- builder.getSymbolRefAttr is gone.
- OpAsmOpInterface's getAsmResultNames method needs explicit...

1dec561cfd7e4370443eab8fb137f981d1a35e9d authored about 3 years ago by Sean Silva <[email protected]>
Update iree-dialects to IREE 7d9e4909f5524e275726b2754d3ad050818d56ae

9cc4fdcaa8f5fe3d65f991ce27e17b22d098063c authored about 3 years ago by Sean Silva <[email protected]>
Refine types continue.

This should cover all the ops that are left in MT.

3b0e5910a8e18a202542d54573fb8ae484058331 authored about 3 years ago by Yi Zhang <[email protected]>
Add sigmoid lowering

Follows existing conventions for activation functions

d9df4bfc9591fcd4225752e24ecbb9e092fde11b authored about 3 years ago by dan <[email protected]>
Delete RestrictedCanonicalizer

It doesn't work properly with the new dialect registration framework.
This was latent and only w...

29e1b2fe89848d58c9bc07e7df7ce651850a5244 authored about 3 years ago by Sean Silva <[email protected]>
fixed some python imports

Change required to enable
./tools/torchscript_e2e_test.sh --config=iree

d7320f3bdaa35f3b6c6104d3949a06fcd5f50a46 authored about 3 years ago by dan <[email protected]>
Revert "Make verbose testing also report compile/trace/run messages."

This reverts commit d8db41b3b6a53138c42b552131b0051050bc9bc1.

These printouts didn't interopera...

1c53424fe70e7a53f67313f172600c310ee67141 authored about 3 years ago by Sean Silva <[email protected]>
Changes to refine types

- Add `!torch.optional` knowledge tracking
- Changes to improve type propagation for branches an...

d6b9709fa5f746b25a3e64304b9c01b648aa9b5e authored about 3 years ago by Yi Zhang <[email protected]>
Add more folders to fold away branches

Added folders to a few binary computing ops, `TupleUnpack`,
`__contains__.str` and `__getitem__....

bc5eae41ca33a2f83026453dff644555c531a2db authored about 3 years ago by Yi Zhang <[email protected]>
Make verbose testing also report compile/trace/run messages.

Helped with #287.

d8db41b3b6a53138c42b552131b0051050bc9bc1 authored about 3 years ago by Stella Laurenzo <[email protected]>
Merge npcomp and mlir python namespaces.

* Now the parts of the MLIR API are directly exported under the npcomp module (i.e. `npcomp.ir`,...

4148f88576a683e2fa530ce146048c025368360b authored about 3 years ago by Stella Laurenzo <[email protected]>
Fix flaky test_export_cat.py lit test (upstream change).

177ccdd55beecbf20b4f8796c2f615c905f36ae3 authored about 3 years ago by Stella Laurenzo <[email protected]>
Integrate llvm-project at a8de667af092c9b4b3b4a95827a521602ebf14ed.

* Requires patch https://reviews.llvm.org/D108527

32f56c67f4240a962103d97754c8c704dcfcac3b authored about 3 years ago by Stella Laurenzo <[email protected]>
Add a few missing deps exposed by stricter linking with BFD.

80ff744c56bb992603fc984010c6bab36f4092b1 authored about 3 years ago by Stella Laurenzo <[email protected]>
Add TorchToIREE and factor out TorchConversion dialect.

This converts a basic list op (torch.prim.ListConstruct) to the IREE
dialect.

```
def forwa...

cab8d922ec11084e3810675fbb9942901f114435 authored about 3 years ago by Sean Silva <[email protected]>
Fix compilation errors from MT model

With the following changes the compilation can continue until
RefineTypes pass:

- Add operators...

85ff8b692b75f1b3b59ef9e52dcac47499465585 authored about 3 years ago by Yi Zhang <[email protected]>
Deal with CMP0116

78fd07da5f303408254a0ac81932eca81aa110d3 authored about 3 years ago by M4tr1xt4ng <[email protected]>
Initial import of iree-dialects.

We plan on using these dialects "natively" as part of the npcomp backend
contract, and provide f...

0b7dbf5f81249734c7004118898d362add14e4b9 authored about 3 years ago by Sean Silva <[email protected]>
Update IREE xfail sets.

All tests pass after https://github.com/google/iree/pull/6666 :)

37df45ded4006c27a66fc1810872dafbbce9bec2 authored about 3 years ago by Sean Silva <[email protected]>
E2E framework: Add support for list/dict/scalar values

Most of the change is in the reporting code to give error messages that
are useful, and adjustin...

6105b0f85169287d69a08b8ea6e92b60df3f7e70 authored about 3 years ago by Sean Silva <[email protected]>
Import Machine Translation model to MLIR.

This includes the following changes to import MT model into MLIR. There
are still a lot of work ...

bfc3ee35c67f56ac7473c52eeee5eb72cde690b7 authored about 3 years ago by Yi Zhang <[email protected]>
Remove npcomp-iree-backend-lower-linkage pass.

This is no longer needed by IREE.

a3bfd115ee758b408c01c001e0da19f253f9ee8b authored about 3 years ago by Sean Silva <[email protected]>
Add resnet inference jupyter notebook.

This takes the example from torchscript_resnet18_e2e.py and puts it into
a slightly cleaned up n...

902c2e579bb089991743e44e3448525c0b494ad1 authored about 3 years ago by Sean Silva <[email protected]>
Triage e2e IREE failures for npcetest.

ResNet works with static shapes. (our test is not static though).

All tests blocked on https://...

f71845ea75fe489ed3f2df5291447b3436d86a07 authored about 3 years ago by Sean Silva <[email protected]>
Add npcomp-lsp-server.

To use, do `ninja npcomp-lsp-server`, copy `build/bin/npcomp-lsp-server`
into your PATH somewher...

c464cb107fff332cee973180b9b783e0ac98c382 authored about 3 years ago by Sean Silva <[email protected]>
Add torch.aten.flatten.using_ints and aten.MaxPool2d linalg lowering

- torch.aten.flatten.using_ints to linalg lowering
- torch.aten.max_pool2d to linalg lowering
- ...

0342b73bf10f3763b356b129d71b68ba0a105ee4 authored about 3 years ago by Yi Zhang <[email protected]>
Rename npcomp-run-mlir to refback-run

This better represents its limited scope. This was causing confusion --
people were feeding it h...

496051163fc517a92cf7805197c9a6a1f1ba803f authored about 3 years ago by Sean Silva <[email protected]>
Minor wordsmithing to README

719f0cd709065fffbcf18e90afb6dfdd3bb012cb authored about 3 years ago by Sean Silva <[email protected]>
Refresh Dockerfile and instructions.

Related to https://github.com/llvm/mlir-npcomp/issues/266

a153cf4ef26216590a95cad5d5600bbbd5e0dda5 authored about 3 years ago by Sean Silva <[email protected]>
Add E2E support for tests with heavy dependencies (heavydep tests).

The tests use the same (pure-Python) test framework as the
normal torchscript_e2e_test.sh, but t...

453e29ea058de3966175ff3d7e445a3699603caf authored about 3 years ago by Sean Silva <[email protected]>
Remove TCF and TCP.

These were legacy concepts that are now superceded by direct Torch to
linalg-on-tensors lowering...

f168cacd6d19f0b11915e3c557a82519fd6753b5 authored about 3 years ago by Sean Silva <[email protected]>
Remove CI pinning.

7c788dbfec010df14f4d225491ab209d3ce181ee authored about 3 years ago by Sean Silva <[email protected]>
Add an e2e test example for Resnet18

Show an example of classifying image from
https://commons.wikimedia.org/wiki/File:YellowLabrador...

93816ee21a89f4331d5610e3a85732dfecc9f854 authored about 3 years ago by Yi Zhang <[email protected]>
Re-enable integration tests in CI.

8494455282674866cff5ce63529dcf0a06eea2b2 authored about 3 years ago by Stella Laurenzo <[email protected]>
Lower accuracy to make e2e pass

`Conv2dNoPaddingModule_basic` and `Conv2dWithPaddingModule_basic` start
failing because of resul...

58b2109898b725b17f47e69978bf36bcbfd6c0ab authored about 3 years ago by Yi Zhang <[email protected]>
Build packages for npcomp-torch.

* Adds a minimal setup.py for frontends/pytorch
* Makes npcomp-core export its headers and libra...

445472c51e6951e22ee6e10e0499d8eb9e672fad authored about 3 years ago by Stella Laurenzo <[email protected]>
Bump llvm-project to 5b2e7f50a6798fd9b9c79d9d62fdebcd9e78525b. (#260)

cd44a35177cc9ae4edd4e3e32a8a22fab7e3a56f authored about 3 years ago by Stella Laurenzo <[email protected]>
Remove mlir-hlo submodule.

dc305c5101a5115baca14a379130e2f1bb7546f4 authored about 3 years ago by Stella Laurenzo <[email protected]>
Add basic setup.py file for the npcomp-core package. (#256)

77e8ecaff8cb7a9b491faad7f3f37bdc36e32d0e authored about 3 years ago by Stella Laurenzo <[email protected]>
Update readme and scripts for setting the new PYTHONPATH

Add scripts for generating .env and update instructions in README.

6fbf94f0b2dcdeb8b27a56daa8b5dd898d59312b authored about 3 years ago by Yi Zhang <[email protected]>
Misc fixes for MacOS. (#255)

* Change aligned_alloc -> malloc. It can fail (and does on MacOS) and is a bit over-aggressive o...

ec611c1e6f44eb5b49c658fd98740000935a1058 authored about 3 years ago by Stella Laurenzo <[email protected]>
Rework the python build to a static assembly of MLIR+NPCOMP (#251)

* Adapt to python build system updates.

* Bump llvm to 310c9496d80961188e8d8f8ad306cdf44bd754...

2dbab50444e9c30eabbd3355a47545c0650fd100 authored about 3 years ago by Stella Laurenzo <[email protected]>
Bump llvm-project to a085c23aa3c8f91866d7f4588d4f683407dc775d. (#250)

* Added additional *ToLLVM conversion patterns (they were disaggregated from standard).
* Misc ...

2ecbcbf8c7b97c6206b2e56ee199f64276a090af authored about 3 years ago by Stella Laurenzo <[email protected]>
Linalg lowering for aten.conv2d and aten.AdaptiveAvgPool2d

1. Add m_TorchConstantIntList
2. Lowering for aten.conv2d
3. Lowering aten.AdaptiveAvgPool2d

89d4931324589bf75ed088e98888ff21fe7cd41e authored over 3 years ago by Yi Zhang <[email protected]>
Bump LLVM version to 7c35aae35b2c386b59af58c56ed36908f3d68371

5f1b2ba3232181551cbe3c31b72c409de0d64bf4 authored over 3 years ago by Yi Zhang <[email protected]>
Bump llvm-project to da289a174fc6617c7be37be2947480510fd4f02a

- Build adjustments for `.cpp.inc` dialect files.
- Renaming of `memref.dim` to `tensor.dim` for...

83b5b5456d3f4125e64e6822713bc36ab73d8273 authored over 3 years ago by Sean Silva <[email protected]>
Add E2E tests to CI

This includes IREE and RefBackend.

This includes a fixup to torchscript_e2e_test.sh for handlin...

ef118eb1e12797ea55ea3f89a9411ad44d4f848b authored over 3 years ago by Sean Silva <[email protected]>
Pin PyTorch version in the CI.

I'm seeing the following error:

```
CMake Error in frontends/pytorch/csrc/CMakeLists.txt:
Imp...

30400d549227033cb7235398352f0c9d62f55e88 authored over 3 years ago by Sean Silva <[email protected]>
Fix README.md

Use the new `tools/torchscript_e2e_test.sh`.

Also, fix a few whitespace/comment issues.

c289d83407951e6d8ec654ebbafb3e4c6e4202f9 authored over 3 years ago by Sean Silva <[email protected]>
Add IREE support in TorchScript e2e tests.

- Add support for "expected failures" in test reporting. The new error
reports look like
[th...

d5108b9dc1fdbd7d9032f7af54175a7f4a6a0eec authored over 3 years ago by Sean Silva <[email protected]>
Generalize support for elementwise ops.

We plumb through e2e a fair number of interesting cases:
- unary, binary, ternary elementwise op...

79928cd2dd6e61573f62aecf47b967b727eee824 authored over 3 years ago by Sean Silva <[email protected]>
Undo CI pinning.

The underlying issue seems to be resolved now:
https://github.com/pytorch/pytorch/issues/60485

577bf1600a3e68685e4512d603fbd4792909f579 authored over 3 years ago by Sean Silva <[email protected]>
Handle rank-0 annotations properly.

49b5b7272bbdc46801826714e57dec5f984fd722 authored over 3 years ago by Sean Silva <[email protected]>
Bump llvm-project to a37cf17834d39411ed1d669098b428f8374c5b45

Changes:
- Change to operand ordering of `linalg.fill`.

145d4ae23c4937472263a8483f7a9313f3534834 authored over 3 years ago by Sean Silva <[email protected]>
Make torch.constant.float print a little nicer.

This printing is chosen to be similar to how MLIR prints the values by
default.

90c6c64fd61b31cc1f023a78a956168bb45c8e58 authored over 3 years ago by Sean Silva <[email protected]>
Add CastOpInterface to torch.prim.unchecked_cast.

This allows it to fold away in trivial cases.

60a947b4a7cc0edd21f00e0cc10e6e3bacaf3299 authored over 3 years ago by Sean Silva <[email protected]>
Add TorchToSCF pass.

1. Add TorchToSCF pass.
2. Convert prim.If and prim.If.yield.

45f2edfc7ac56e3886803b1280b345d0454dc448 authored over 3 years ago by Yi Zhang <[email protected]>
More folding for aten.gt.int, aten.ne.int and Aten__Getitem__TOp.

- Fold more for aten.gt.int, aten.ne.int and Aten__Getitem__TOp
- Some format cleaning up

5ad144c4feadcf88aad0f5f5acfa8a19c1a45522 authored over 3 years ago by Yi Zhang <[email protected]>
Make MaximizeValueSemantics a bit smarter.

This adds a pattern to MaximizeValueSemantics which does a simple
abstract interpretation within...

79aade33da54e9c104f8b2024d003024aa63d68c authored over 3 years ago by Sean Silva <[email protected]>
Add torch.aten.batch_norm Linalg lowering support

1. Added a simplified version of torch.aten.batch_norm which only handles
inference and assumes ...

6dddb4d4fe206f7007a6cf693700d842a2094933 authored over 3 years ago by Yi Zhang <[email protected]>
Try again to pin the CI to a working PyTorch version.

For some reason, pytorch_nightly was being installed for the LLVM build,
and so the wrong line g...

bbd749620e4c834a3393df9314183a3cb6e1ccac authored over 3 years ago by Sean Silva <[email protected]>
Pin torch to a specific version in the CI.

This temporarily works around the CMake error:

```
CMake Error in frontends/pytorch/csrc/CMakeL...

f7ebd870f6f84dd7f593546d44898010be08000d authored over 3 years ago by Sean Silva <[email protected]>
Convert Torch constant ops to std.constant

e6adecac834dccd6ed25b54e2b9142aca0b34673 authored over 3 years ago by Yi Zhang <[email protected]>
Make `torch.copy.tensor` canonicalization a bit smarter.

This removes most of the trivial cases that MaximizeValueSemantics needs
to handle, making it ea...

78d2cc08189d861c09425f5ca1982ccc8b19611e authored over 3 years ago by Sean Silva <[email protected]>
Bump llvm-project to 116841c623747972d0ae80239d3ea7b8409b868b

This brings in a change to canonicalization of `dim` ops, which we need
to adjust our pass pipel...

1bc889130d506ddcf1bec913f02197e2e726f86e authored over 3 years ago by Sean Silva <[email protected]>