Skip to content

Commit 6edc827

Browse files
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: update
1 parent 2118c23 commit 6edc827

File tree

6 files changed

+379
-229
lines changed

6 files changed

+379
-229
lines changed

src/enzyme_ad/jax/Dialect/EnzymeXLAOps.td

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -397,7 +397,7 @@ def CommRegionOp : EnzymeXLA_Op<"comm_region", [
397397

398398
// Linear Algebra Ops
399399

400-
def SymmOp: EnzymeXLA_Op<"lapack.symm", [Pure, SameOperandsAndResultElementType]> {
400+
def SymmOp: EnzymeXLA_Op<"blas.symm", [Pure, SameOperandsAndResultElementType]> {
401401
let summary = "Multiplication involving a symmetric matrix";
402402

403403
let description = [{

0 commit comments

Comments
 (0)