EnzymeAD / EnzymeAD/Enzyme-JAX

Select to concat or dynamic_update_slice

Open
#703 3 comments 0 reactions 1 assignee Claimed by @chelini View on GitHub
Dominant language
MLIR
Stars
131
Forks
53
Avg merge
1d 10h
Merged PRs (30d)
193

Description

```
%c_118 = stablehlo.constant dense<"0xFFFFFFFFFFFFFFFFFFFF00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000"> : tensor<8x80xi1>

%5329 = stablehlo.select %c_118, %5324, %5328 : tensor<8x80xi1>, tensor<8x80xf64>

```

```
%c_121 = stablehlo.constant dense<"0x00000000000000000000FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF"> : tensor<8x80xi1>
%5278 = stablehlo.select %c_121, %5277, %cst_122 : tensor<8x80xi1>, tensor<8x80xf64>

%c = stablehlo.constant dense<"0x0000000000000000008000000000000000000080000000000000000000800000000000000000008000000000000000000080000000000000000000800000000000000000008000000000000000000080"> : tensor<8x80xi1>
%5318 = stablehlo.select %c, %5315, %5317 : tensor<8x80xi1>, tensor<8x80xf64>

%c_18 = stablehlo.constant dense<[false, true, false]> : tensor<3xi1>
%4241 = stablehlo.broadcast_in_dim %c_18, dims = [0] : (tensor<3xi1>) -> tensor<3x8x80xi1>
%4285 = stablehlo.select %4241, %4284, %3668 : tensor<3x8x80xi1>, tensor<3x8x80xf64>

%c_20 = stablehlo.constant dense<"0x0100000000000000000001000000000000000000010000000000000000000100000000000000000001000000000000000000010000000000000000000100000000000000000001000000000000000000"> : tensor<8x80xi1>
%5343 = stablehlo.select %c_20, %5342, %5335 : tensor<8x80xi1>, tensor<8x80xf64>

%c_21 = stablehlo.constant dense<"0x00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000FFFFFFFFFFFFFFFFFFFF"> : tensor<8x80xi1>
%5330 = stablehlo.select %c_21, %5323, %5329 : tensor<8x80xi1>, tensor<8x80xf64>

%c_121 = stablehlo.constant dense<"0x00000000000000000000FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF"> : tensor<8x80xi1>

%5278 = stablehlo.select %c_121, %5277, %cst_122 : tensor<8x80xi1>, tensor<8x80xf64>
```

We should be able to get rid of these

@chelini @tyb0807 @hhkit @arpitj1 can you take a look?

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.