Commit 6edc827
committed
fix: use lapack instead of lapacke
feat: get_dimension_size batch interface
feat: implement jitcall batching with shlo_generic_batch_op_interface
refactor: reuse batching interface for LU factorization
fix: remove old changes
feat: mark memory effects
fix: update to new API
fix: use correct return
fix: rework svd lowering to use lapack and workspace buffers
fix: use func return
fix: leading dimension
fix: use backend config
fix: remove unused vars
fix: tpu lowering
test: update1 parent 2118c23 commit 6edc827
File tree
6 files changed
+379
-229
lines changed- src/enzyme_ad/jax
- Dialect
- Passes
- test/lit_tests/linalg
6 files changed
+379
-229
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
397 | 397 | | |
398 | 398 | | |
399 | 399 | | |
400 | | - | |
| 400 | + | |
401 | 401 | | |
402 | 402 | | |
403 | 403 | | |
| |||
0 commit comments