-
Notifications
You must be signed in to change notification settings - Fork 34
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Remove arrays of scaling factors #116
Conversation
Codecov ReportPatch coverage:
Additional details and impacted files@@ Coverage Diff @@
## master #116 +/- ##
==========================================
+ Coverage 93.57% 93.87% +0.30%
==========================================
Files 5 5
Lines 420 441 +21
==========================================
+ Hits 393 414 +21
Misses 27 27
☔ View full report in Codecov by Sentry. |
What do you think of adding tests of view inputs? E.g. see #117 (made a PR to this branch rather than a code suggestion since I cannot code suggest on unchanged files). |
Bump on merge? |
Can we get a release, mainly in order to include this PR and fix Zygote tests? (FluxML/Zygote.jl#1452) |
Alternative to #114 that avoids the computation of the array of scaling factors completely.
Tested locally with regular arrays and static arrays. Using
broadcast
instead ofmap
(potentially includingd
etc. as broadcasted arguments) would returnSizedArray
s instead ofStaticArray
s forSArray
inputs, hence I went withmap
instead.Has to be tested with Zygote and CUDA as well (for the latter it might be useful to set up GPU tests with buildkite and for the former downstream tests with Zygote, limited to FFTs, would be useful).
Edit: Tests of Zygote#master pass locally with this PR.