Avik Pal issues

Results 210 issues of


                                            Avik Pal

Simplify Reduce(Reduce(...))

```mlir module @reactant_fn attributes {mhlo.num_partitions = 1 : i64, mhlo.num_replicas = 1 : i64} { func.func @main(%arg0: tensor) -> tensor { %cst = stablehlo.constant dense : tensor %0 = stablehlo.multiply...

Generalize Batch-Op

```mlir module @reactant_batched... attributes {mhlo.num_partitions = 1 : i64, mhlo.num_replicas = 1 : i64} { func.func private @"Reactant.TracedUtils.TypeCast{Float32}()_broadcast_scalar"(%arg0: tensor) -> tensor { return %arg0 : tensor } func.func private @"Reactant.TracedUtils.TypeCast{Float32}()_broadcast_scalar_1"(%arg0:...

Automatically rewrite `WhileOp` with `batch` if possible

```mlir module @reactant_can_be_... attributes {mhlo.num_partitions = 1 : i64, mhlo.num_replicas = 1 : i64} { func.func @main(%arg0: tensor) -> tensor { %cst = stablehlo.constant dense : tensor %c = stablehlo.constant...

Unused loop iterarg returning constant

``` %iterArg_106 = %15 return %cst_53 ```

Slice access analysis to eliminate slices and concats inside loops

### case I ```mlir %235 = stablehlo.slice %iterArg_101 [0:1, 1:2033, 0:4080] {mhlo.sharding = "{devices=[1,4,8]

Push reshapes up

- [ ] reshape( slice ) -> slice ( reshape ) ~(@jumerckx, https://github.com/EnzymeAD/Enzyme-JAX/pull/581)~ ``` %3250 = stablehlo.reshape %125 {mhlo.sharding = "{devices=[2,1,2,2]

Transpose Optimizations Missing (Move Transposes Up)

[test_sharded.mlir.txt](https://github.com/user-attachments/files/19528762/test_sharded.mlir.txt) - [x] transpose (slice x) -> slice (transpose ) (in https://github.com/EnzymeAD/Enzyme-JAX/pull/567) - [x] transpose reshape - [x] transpose (binary op) -- several of these will get fused with broadcast_in_dims...

Avik Pal