Skip to content

Commit

Permalink
fix crash in gpu mode with archetype legendre priors
Browse files Browse the repository at this point in the history
  • Loading branch information
Stephen Bailey authored and Stephen Bailey committed Jan 3, 2024
1 parent 3a8e5cd commit 8133f74
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion py/redrock/zscan.py
Original file line number Diff line number Diff line change
Expand Up @@ -689,7 +689,7 @@ def calc_zchi2_batch(spectra, tdata, weights, flux, wflux, nz, nbasis, solve_mat
#all_M = calc_batch_dot_product_3d3d_gpu(cp.ascontiguousarray(Tbs.swapaxes(-2, -1)), (weights[None, :, None] * Tbs))
#all_y = (Tbs.swapaxes(-2, -1) @ wflux)
if prior is not None:
all_M += prior
all_M += cp.asarray(prior)

#3) Use new helper method solve_matrices to use appropriate method
#for this template to solve for zcoeff in batch for all_M and all_y.
Expand Down

0 comments on commit 8133f74

Please sign in to comment.