Wu Yuan
|
cc28d566ff
|
[Stablehlo] Support AtenTrilOp (#3359)
1. lower aten.tril to stablehlo composed by iota, select and so forth
2. add related e2e test cases
|
2024-05-20 15:49:24 +08:00 |
Yuanqiang Liu
|
5f7cb9e253
|
[Stablehlo] lowering aten.randn & aten.normal_functional to mhlo.rng … (#3328)
…NORMAL
* split lowering of uniform, randn, normal from Basic.cpp into Rng.cpp
|
2024-05-11 15:33:37 +08:00 |
Yuanqiang Liu
|
53299eb224
|
[Stablehlo] Bump stablehlo to ab92adeda9119a6c3914cd42367b0a2b70765e91 (#3285)
|
2024-05-05 19:56:12 +08:00 |
penguin_wwy
|
122eb69a98
|
[stablehlo] add aten left/right shift op conversion support (#3234)
|
2024-04-26 09:20:49 +08:00 |
Rob Suderman
|
135c81a416
|
[torch] Add folder for `prim.NumToTensor.Scalar` (#2921)
Useful for `slice` lowerings that depend on tensors made form scalars.
|
2024-02-19 11:55:54 -08:00 |
Yuanqiang Liu
|
d778950f45
|
[Torch Dialect] add fold pattern for aten.clone (#2804)
|
2024-01-31 09:43:21 +08:00 |
Aart Bik
|
0aed231e21
|
[torch-mlir][conversion-test] cleanup trailing whitespace in mlir files (#2807)
|
2024-01-25 14:24:28 -08:00 |
Zhekun Zhang
|
f0b7b63be0
|
[Stablehlo] Add aten.uniform lowering (#2101)
* add uniform stablehlo lowering
* add unit test
* new line
* rm redundant file
* Empty commit, trigger test
* fix include
* address comments
---------
Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
|
2023-05-25 10:32:55 +08:00 |
Yuanqiang Liu
|
c86f46bd70
|
[test] rename TorchToMhlo to TorchToStablehlo (#1995)
|
2023-04-03 18:41:25 -07:00 |