Conversation
Removed temporary pin for jax package version.
|
cc @EiffL |
|
arf |
|
Ok, some tests are failing, is it just a matter of re-exporting random numbers for the new JAX RNG implementation? If so, I can do that. |
Merging this PR will degrade performance by 36.8%
Performance Changes
Comparing Footnotes |
|
Yes. Also there are some xfail tests that appear to be fixed with the latest galsim. We should figure out which version that was, use that as the galsim lower bound, and remove the xfail marker. |
|
Cool, on it, I'll push here, and try to make this branch pass the tests |
|
Ok this first batch of fixes above addresses the following:
some additional fixes remain to apply |
|
This last commit fixes the random noise test with a local hook that overrides the hard coded values. If this works we should open a separate PR for galsim to update the following values: if str(module_path).endswith("tests/GalSim/tests/test_random.py"):
module.obj.uResult = (0.0303194914, 0.0910759047, 0.1208923360)
module.obj.gResult = (-1.3035798312, 0.4306917482, 0.9542795210)
module.obj.bResult = (7, 6, 7)
module.obj.pResult = (5, 8, 6)
module.obj.wResult = (3.7699892848, 5.0030654033, 5.3921485618)
module.obj.gammaResult = (0.7985896238, 22.0508132116, 33.1369864688)
module.obj.chi2Result = (19.2174896025, 47.3448788104, 55.8177548146)But this can be done separately. |
|
ok, only 2 failing tests left. |
There was a problem hiding this comment.
this will become obsolete once the PR in galsim is merged, and we can remove it then
There was a problem hiding this comment.
I don't understand this comment. We still need some of the fixtures here even if the PR on galsim is merged.
There was a problem hiding this comment.
oh sorry, I mean these edits I made were redundant with the values I later on updated directly on Galsim
|
Alright, I've made modifs on the GalSim repo here and temporarily changed the pointer of the submodule to my branch: GalSim-developers/GalSim#1344 NOTE: We should revert back the submodule pointer to |
Co-authored-by: Matthew R. Becker <beckermr@users.noreply.github.com>
|
I've switched the submodule back to point to the |
|
@EiffL the module level override for the test values is a pretty good. Should we switch to that to make future updates easier and less intrusive in galsim? |
|
I mention it because the rng results are less stable than the other changes. |
|
I think that would be a good idea, but that requires some refactoring of the current test modifications in galsim. I couldn't use the hooks everywhere because they only work on module level variables, but a number of parameters are hard coded inside functions and we can't access them. So I think it would be a wortwhile endeavor, but an additional/seperate task; we only need to change the galsim tests no modifications are needed here, until the next time JAX changes their RNG |
We may need to revisit the RNG thing.