EnzymeAD / EnzymeAD/Enzyme-JAX
DUS of DUS -> DUS of extend
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
```
dyn: %5179 = stablehlo.dynamic_update_slice %5171, %5168, %c_354, %c_353, %c_356 : (tensor<20x1536x3056xf32>, tensor<4x1x3056xf32>, tensor, tensor, tensor) -> tensor<20x1536x3056xf32>
+ op: %5171 = stablehlo.dynamic_update_slice %5146, %5165, %c_354, %c_355, %c_356 : (tensor<20x1536x3056xf32>, tensor<4x1x3056xf32>, tensor, tensor, tensor) -> tensor<20x1536x3056xf32>
+ + op: %5146 = stablehlo.dynamic_update_slice %105, %5145, %c_355, %c_354, %c_356 : (tensor<20x1536x3056xf32>, tensor<6x1520x3056xf32>, tensor, tensor, tensor) -> tensor<20x1536x3056xf32>
+ + up: %5165 = stablehlo.slice %5085 [0:4, 0:1, 0:3056] : (tensor<4x1520x3056xf32>) -> tensor<4x1x3056xf32>
+ up: %5168 = stablehlo.slice %5085 [0:4, 1519:1520, 0:3056] : (tensor<4x1520x3056xf32>) -> tensor<4x1x3056xf32>
```
this can instead be a dus of an extend
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.