EnzymeAD / EnzymeAD/Enzyme-JAX

Simplify comparison

Open
#811 4 comments 0 reactions 0 assignees View on GitHub
Dominant language
MLIR
Stars
131
Forks
53
Avg merge
1d 10h
Merged PRs (30d)
193

Description

Simplifying comparisons

((a <= c1) && cond2) && (a <= c2).

this should be come (a <= min(c1, c2)) && cond2
```
%3249 = stablehlo.compare LE, %3241, %c_304 {mhlo.sharding = "{devices=[1,4,4]<=[4,4]T(1,0)}"} : (tensor<4x6128x12272xi64>, tensor<4x6128x12272xi64>) -> tensor<4x6128x12272xi1> loc(#loc2176)
%3250 = stablehlo.compare LE, %3242, %c_304 {mhlo.sharding = "{devices=[1,4,4]<=[4,4]T(1,0)}"} : (tensor<4x6128x12272xi64>, tensor<4x6128x12272xi64>) -> tensor<4x6128x12272xi1> loc(#loc2176)
%3251 = stablehlo.compare LE, %3243, %c_304 {mhlo.sharding = "{devices=[1,4,4]<=[4,4]T(1,0)}"} : (tensor<4x6128x12272xi64>, tensor<4x6128x12272xi64>) -> tensor<4x6128x12272xi1> loc(#loc2176)
%3252 = stablehlo.compare LE, %3248, %c_304 {mhlo.sharding = "{devices=[1,4,4]<=[4,4]T(1,0)}"} : (tensor<4x6128x12272xi64>, tensor<4x6128x12272xi64>) -> tensor<4x6128x12272xi1> loc(#loc2176)
%3253 = stablehlo.compare GE, %3241, %c_308 {mhlo.sharding = "{devices=[1,4,4]<=[4,4]T(1,0)}"} : (tensor<4x6128x12272xi64>, tensor<4x6128x12272xi64>) -> tensor<4x6128x12272xi1> loc(#loc2176)
%3254 = stablehlo.compare GE, %3247, %c_308 {mhlo.sharding = "{devices=[1,4,4]<=[4,4]T(1,0)}"} : (tensor<4x6128x12272xi64>, tensor<4x6128x12272xi64>) -> tensor<4x6128x12272xi1> loc(#loc2176)
%3255 = stablehlo.and %3249, %3253 {mhlo.sharding = "{devices=[1,4,4]<=[4,4]T(1,0)}"} : tensor<4x6128x12272xi1> loc(#loc2176)
%3256 = stablehlo.compare LE, %3241, %c_306 {mhlo.sharding = "{devices=[1,4,4]<=[4,4]T(1,0)}"} : (tensor<4x6128x12272xi64>, tensor<4x6128x12272xi64>) -> tensor<4x6128x12272xi1> loc(#loc2336)
%3257 = stablehlo.compare LE, %3242, %c_306 {mhlo.sharding = "{devices=[1,4,4]<=[4,4]T(1,0)}"} : (tensor<4x6128x12272xi64>, tensor<4x6128x12272xi64>) -> tensor<4x6128x12272xi1> loc(#loc2336)
%3258 = stablehlo.compare LE, %3243, %c_306 {mhlo.sharding = "{devices=[1,4,4]<=[4,4]T(1,0)}"} : (tensor<4x6128x12272xi64>, tensor<4x6128x12272xi64>) -> tensor<4x6128x12272xi1> loc(#loc2336)
%3259 = stablehlo.compare LE, %3247, %c_306 {mhlo.sharding = "{devices=[1,4,4]<=[4,4]T(1,0)}"} : (tensor<4x6128x12272xi64>, tensor<4x6128x12272xi64>) -> tensor<4x6128x12272xi1> loc(#loc2336)
%3260 = stablehlo.compare LE, %3248, %c_306 {mhlo.sharding = "{devices=[1,4,4]<=[4,4]T(1,0)}"} : (tensor<4x6128x12272xi64>, tensor<4x6128x12272xi64>) -> tensor<4x6128x12272xi1> loc(#loc2336)
%3261 = stablehlo.and %3255, %3256 {mhlo.sharding = "{devices=[1,4,4]<=[4,4]T(1,0)}"} : tensor<4x6128x12272xi1> loc(#loc2176)

```

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.