Tanyo Kwok

[MHLO] Init end-to-end unit tests

8

See RFC https://github.com/llvm/torch-mlir/issues/999 Co-authored-by: Bairen Yi [email protected] Co-authored-by: Jiawei Wu [email protected] Co-authored-by: Tianyou Guo [email protected] Co-authored-by: Xu Yan [email protected] Co-authored-by: Ziheng Jiang [email protected]

Add decomposition for aten.flatten.using_ints

8

Can you review for me @silvasean @powderluv. Thanks!

How to lower the pattern `torch.aten.Int.Scalar(torch.aten.item(rank0_tensor))`?

1

I want to lower `torch.dialect` module something like the following: ```mlir module { func @main(%arg0: !torch.vtensor, %arg1: !torch.vtensor) -> !torch.vtensor { %int1 = torch.constant.int 1 %int3 = torch.constant.int 3 %int2...

[RFC] Add TorchBlade benchmarks based on TorchBench

5

### Why TorchBench? 1. A benchmark maintained by the PyTorch community officially 2. `torchbenchmark/models` contains copies of popular or exemplary workloads 3. `torch.utils.benchmark.Timer` has priorities than `timeit.Timer` in PyTorch 4....

Please add mhlo::PadOp lowering in disc

2

Currently, DISC lowers all pad-like op into `mhlo::DynamicPadOp`. However, the `DPadToPad` pattern in `mhlo::DynamicPadOp`'s CanonicalizationPatterns may convert `mhlo::DynamicPadOp` into `mhlo::PadOp` since the rebase https://github.com/alibaba/BladeDISC/pull/525. It will cause lowering exception of...

The dump toolkit dumps the transformer training graph based on LTC, next we should implement following op converters: - [ ] aten::_log_softmax - [ ] aten::_log_softmax_backward_data - [ ] aten::_softmax...

Completing the Torch2MHLO operator conversion

1

Background: We are working on building Torch2MHLO passes based on [TorchMLIR](https://github.com/llvm/torch-mlir). The goal for the most recent is to switch from the old converter framework to TorchMLIR: - [ ]...