Use Testsuite for AD tests #126
Conversation
|
Your PR no longer requires formatting changes. Thank you for your contribution! |
kshyatt
force-pushed
the
testsuite-ad
branch
2 times, most recently
from
0669d08 to
958dd36
Compare
kshyatt
force-pushed
the
testsuite-ad
branch
6 times, most recently
from
fad3f4a to
daeab4d
Compare
|
I think this is ready for review if CI passes, it's not testing the Diagonal stuff for now waiting on @lkdvos 's PR, but it will radically simplify adding tests for that and CUDA and Enzyme |
lkdvos
left a comment
lkdvos
left a comment
There was a problem hiding this comment.
I left a bunch of smaller comments through the code, and only quickly went through the actual test suites. Am I correct that these haven't changed much?
I think in general it would be nice to try and link some of the GPU issues we are facing with actual issue numbers and resolve them through dispatch alternatives, rather than having to alter the code for regular arrays as well.
In particular, isa(A, Array) seems a bit to strict since we also support views etc, which would then also take the slow path.
| :svd_compact, :svd_trunc, :svd_trunc_no_error, :svd_vals, | ||
| :left_polar, :right_polar, | ||
| ) | ||
| copy_f = Symbol(:cr_copy_, f) |
There was a problem hiding this comment.
Why is this cr_ ? what does that mean?
There was a problem hiding this comment.
I think I just got the difference with the mc_ ones, I thought this should all be run in separate modules though so there shouldn't necessarily be any nameclashing here?
I might be missing something though
There was a problem hiding this comment.
Yes they might be, I gave them different names just to be super clear but I guess it is not so clear 😓
|
Now that the enzyme stuff is merged, I'll work on incorporating that into this PR and addressing the remaining comments |
|
OK, the Enzyme tests are now part of the testsuite. I turned off the |
This lets us reuse a lot of the setup infrastructure for ChainRules, Mooncake, and (soon) Enzyme. Also starts testing the AD rules on GPU.