torch-mlir/e2e_testing/torchscript
Yi Zhang 0cb216a1ad [Torch][Linalg] Add basic support for RNG
This PR include the following pieces:
- Add torch `Generator` type. `Generator` type is converted to i64 in
refbackend type converter.
- Add seed managment support for the default global generator.
`torch_c.getNextSeed` op is used to get the seed. On refbackend, the
`torch_c.getNextSeed` is lowered to load/store from [0] of global
variable `default_generator` memref<i64> in `InsertRngGlobals` pass.
- Add `aten.uniform_` and testing as an example op for RNG ops. Add
`torch.pseudo.aten.uniform` op. It has the same operands and return as
the `aten.uniform_` from the op registry except for value semantics.
2022-01-31 18:56:42 -05:00
..
arange.py [MLIR][TORCH] Add E2E support for torch.arange op 2021-12-27 22:45:48 +05:30
argmax.py add argmax lowering 2021-10-13 14:31:16 -04:00
backprop.py Add e2e test for aten.log_softmax_back_data op 2021-11-19 00:08:28 +05:30
basic.py [TORCH] Add aten.std e2e support 2022-01-31 15:17:49 -05:00
batchnorm.py [MLIR][TORCH] Add E2E support for `aten.native_layer_norm`. (#470) 2021-12-10 19:06:19 +05:30
constant_alloc.py [TORCH][MLIR] Add E2E support for `aten.[ones_like|zeros_like]` 2022-01-06 20:24:40 +05:30
conv.py Refine static shapes for conv2d and maxpool2d 2022-01-03 11:09:23 -06:00
elementwise.py Make broadcasting result shape more static 2022-01-06 18:39:27 -05:00
histogram_binning_calibration.py Adding an e2e test for histogram binning calibration 2022-01-25 18:27:20 -05:00
index_select.py [MLIR][TORCH] Add E2E support for aten.index_select op 2021-12-09 23:13:36 +05:30
main.py [Torch][Linalg] Add basic support for RNG 2022-01-31 18:56:42 -05:00
matmul.py Add lowering of aten.matmul op. 2021-10-26 12:45:09 -04:00
mlp.py Support aten::linear with rank 3 inputs 2021-11-18 22:15:04 +05:30
nll_loss.py * Workaround for Issue 521, remove createTosaToStandard from Passes.cpp and 2022-01-12 14:30:58 -06:00
quantized_models.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
reduction.py Add lowering of `aten.max.dim` op. 2022-01-31 21:41:22 +05:30
rng.py [Torch][Linalg] Add basic support for RNG 2022-01-31 18:56:42 -05:00
scalar.py add aten.sub.int/aten.mul.int lowering in TorchToStd 2021-12-17 10:35:15 -08:00
slice_like.py Small fixes for slice edge cases (#476) 2021-12-15 15:54:41 -06:00
squeeze.py [TORCH][MLIR] Add E2E support for `aten.squeeze.dim` op 2021-12-10 17:01:20 +05:30
table_batch_embedding.py [TBE] Add a test module for table batch embedding 2022-01-28 02:24:28 +05:30
threshold.py [MLIR][TORCH] Add E2E support for aten.threshold, aten.threshold_backward op 2022-01-10 11:56:56 +05:30
type_conversion.py Add convertScalarToDtype helper. 2021-11-08 17:50:52 -05:00
type_promotion.py Add scalar type promotion for mul and div (#454) 2021-12-03 13:51:25 -06:00
view.py [TORCH][MLIR] Fold trivial cases of `aten.to.dtype` and `aten.view` op 2021-12-24 13:32:34 +05:30
vision_models.py Add min/max/clamp support. 2021-10-27 13:29:21 -07:00
xfail_sets.py [tosa] Add maxpool2d and adaptive_avgpool2d support (#550) 2022-01-31 13:34:09 -08:00