Skip to content

Add Mooncake and Enzyme tests for Diagonal#179

Open
kshyatt wants to merge 1 commit intomainfrom
ksh/diag
Open

Add Mooncake and Enzyme tests for Diagonal#179
kshyatt wants to merge 1 commit intomainfrom
ksh/diag

Conversation

@kshyatt
Copy link
Copy Markdown
Member

@kshyatt kshyatt commented Feb 27, 2026

No description provided.

@kshyatt kshyatt requested a review from lkdvos February 27, 2026 10:24
@codecov
Copy link
Copy Markdown

codecov Bot commented Feb 27, 2026

Codecov Report

❌ Patch coverage is 45.34884% with 47 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
...ixAlgebraKitEnzymeExt/MatrixAlgebraKitEnzymeExt.jl 0.00% 31 Missing ⚠️
...gebraKitMooncakeExt/MatrixAlgebraKitMooncakeExt.jl 70.90% 16 Missing ⚠️
Files with missing lines Coverage Δ
...gebraKitMooncakeExt/MatrixAlgebraKitMooncakeExt.jl 63.93% <70.90%> (-0.39%) ⬇️
...ixAlgebraKitEnzymeExt/MatrixAlgebraKitEnzymeExt.jl 0.00% <0.00%> (-1.30%) ⬇️

... and 4 files with indirect coverage changes

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@kshyatt kshyatt changed the title Add Mooncake tests for Diagonal Add Mooncake and Enzyme tests for Diagonal Mar 26, 2026
@kshyatt
Copy link
Copy Markdown
Member Author

kshyatt commented Mar 26, 2026

Went a little crazy with this one after figuring out some of the rules needed modification for the Diagonal case where the output is by default the same as the input

@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented Mar 26, 2026

Your PR no longer requires formatting changes. Thank you for your contribution!

@kshyatt kshyatt marked this pull request as ready for review March 27, 2026 16:59
@kshyatt
Copy link
Copy Markdown
Member Author

kshyatt commented Mar 27, 2026

This was a fun one, as a bunch of rules had to be modified to account for the case where one of the arguments === the input. Seems to all be working now, at least!

Copy link
Copy Markdown
Member

@lkdvos lkdvos left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Good catch, that does indeed sound like an absolute misery to debug/find and solve!

I left some small suggestions around code organization, mostly to reduce the amount of code and/or improve readability, but otherwise this looks great (I absolutely understand that the complexity might just be unavoidable, again, impressive catch on this!)

I'll go through the remainder afterwards if you like, but I will already leave this here to not get lost in a wall of questions :)

Comment thread ext/MatrixAlgebraKitEnzymeExt/MatrixAlgebraKitEnzymeExt.jl Outdated
Comment thread ext/MatrixAlgebraKitEnzymeExt/MatrixAlgebraKitEnzymeExt.jl Outdated
Comment thread ext/MatrixAlgebraKitEnzymeExt/MatrixAlgebraKitEnzymeExt.jl
Comment thread ext/MatrixAlgebraKitMooncakeExt/MatrixAlgebraKitMooncakeExt.jl
Comment thread ext/MatrixAlgebraKitEnzymeExt/MatrixAlgebraKitEnzymeExt.jl Outdated
@kshyatt
Copy link
Copy Markdown
Member Author

kshyatt commented Mar 31, 2026

CUDA fail is unrelated, it's the CUSOLVER memory thing again

Fix stupid typos

Diagonal + eig(h) working

SVD working for diag

LQ/QR working

Polar and orthnull

Some Mooncake progress

Formatter

Working Mooncake eig

Remove comment

No diag for orthnull

Diag tests working for Mooncake

Formatter

Update ext/MatrixAlgebraKitEnzymeExt/MatrixAlgebraKitEnzymeExt.jl

Co-authored-by: Lukas Devos <[email protected]>

A bit of Enzyme cleanup

A little Mooncake cleanup
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants