77a68fc52f
test examples for multi tensor const folding ( #4031 )
...
works with literal const operand now because it's copied to each shard and handled by lazy.
does not work for sharded const
2024-04-01 16:53:43 -04:00
379d52548d
const fold left const operand for ADD and MUL ( #4029 )
...
* const fold left const operand for ADD and MUL
* neg have dtype issue
2024-04-01 15:09:04 -04:00
0e02d074bd
fix Tensor.pow folding for exponent 0 and 1 ( #4025 )
2024-03-31 19:57:23 -04:00
d3f27761b0
move const folding of ADD/SUB/MUL from tensor to lazy ( #4020 )
...
* move const folding of ADD/SUB/MUL from tensor to lazy
will do div and pow separately.
* fix onnx adding with None
2024-03-31 16:35:36 -04:00
7f859593b8
fix _to_const_val and const folding around it ( #4017 )
...
* fix _to_const_val and const folding around it
is_unrealized_contiguous_const is too strict and almost never hit if const is expanded.
suffice to check if there's no pad
* that test is folded
* test_const_folding
2024-03-31 13:09:23 -04:00
chenyu