-
Notifications
You must be signed in to change notification settings - Fork 13
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
GPU acclelerated archetypes.get_best_archetype().
Use batch rebin, transmission_Lyman, and calc_zchi2 operations. Speed gains: without archetypes, redrock on 4 GPU / 4 CPU takes 14.8s reported total run time, 7.3s of which is in the fine redshift scan. Comparatively with 64 CPU and 0 GPU the base redrock runs in 40.0s reported total run time with 6.7s spent in fine redshift scan. Adding the base archetypes option (without per-camera or nearest neighbor) raises CPU times to 63.8s overall and 28.1s in fine z scan so about 60% increase overall and about 4x slower in fine z scan. With the new code the "batch" CPU mode slightly improves this to 60.0s and 24.2s. With the new GPU code, it runs on 4 GPU / 4 CPU in 22.8s total and 14.3s for fine z scan, a 50% overall increase but only 2x speed increase in the fine z scan, a big improvement from CPU times. Also updated transmission_Lyman to return None when given scalar z to match behavior when given an array of redshifts, in the case where the wavelength range is not affected by Lyman transmission. There is no need in this case to calculate an array of all ones and then additionally multiply the rebinned data by it. Have yet to update --per-camera and -n_nearest options so right now there are placeholders so that it does not crash that simply loop over the existing CPU-mode logic. These will be updated shortly.
- Loading branch information
1 parent
d0395f5
commit 055aa12
Showing
4 changed files
with
123 additions
and
26 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters