EnzymeAD / EnzymeAD/Enzyme-JAX

support dynamic-size matrices on lowering of `lapack` routines

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

Description

currently, we generate one LAPACK LLVM function wrapper per instance because we hardcode the sizes of the matrices on it. this could be better handled by just passing the sizes as arguments to the `jit_call`.

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by tracing the lowering of LAPACK routines to the generated LLVM function wrappers and the jit_call path. Replace hardcoded matrix sizes with size arguments passed to jit_call, and verify that dynamic-size matrices no longer require one wrapper per instance.

Written by the indexing model from the issue text.

Assessment

Domain
compilers
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
38/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.