EnzymeAD / EnzymeAD/Enzyme-JAX

Pass to convert top_k with k = 1 to a reduction

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

Description

See https://github.com/EnzymeAD/Reactant.jl/pull/1439

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by reading the linked Reactant.jl pull request, since the issue body provides no repository file, test, or entry point. Trace how top_k with k = 1 is represented and converted there; done means the corresponding conversion pass is implemented and its existing tests cover the reduction result.

Written by the indexing model from the issue text.

Assessment

Domain
compilers
Issue type
Refactor
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
15/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.