EnzymeAD/Enzyme-JAX

`x != false` and others

Open

#1,417 opened on Sep 23, 2025

View on GitHub
 (5 comments) (0 reactions) (0 assignees)MLIR (45 forks)auto 404
good first issue

Repository metrics

Stars
 (126 stars)
PR merge metrics
 (PR metrics pending)

Description

%4 = stablehlo.compare  NE, %3, %c_85,  UNSIGNED : (tensor<8000x3xi1>, tensor<8000x3xi1>) -> tensor<8000x3xi1>

This actually shows up in jaxmd 😅

Contributor guide

Research direction
Examine the generated stablehlo code for 'x != false' to understand why the result is a tensor of i1 instead of the expected type. Check if this is a known pattern in the JAX to MLIR lowering.
Tech stack
cpp
Domain
compilers
Issue type
Bug
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
2
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
1-3 hours
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
Active
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
Needs investigation
Prerequisites
GitMLIR
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
30