Commit 1f43161c authored by Michiel Cottaar's avatar Michiel Cottaar
Browse files

BUG: don't del bakward_idx if only_forward is set

parent 2b5e6a04
Pipeline #5292 failed with stage
in 9 minutes and 53 seconds
......@@ -609,7 +609,6 @@ __global__ void matrix_mult_nd_invert({dtype} *derparam, {dtype} *derfield, int
def update_indices(self, only_forward=False):
del self.forward_idx
del self.backward_idx
if only_forward:
forward_compressed, forward_idx = self.basis.within_range(self.request, return_compressed=True)
else:
......@@ -621,6 +620,7 @@ __global__ void matrix_mult_nd_invert({dtype} *derparam, {dtype} *derfield, int
if only_forward:
return
del self.backward_idx
backward_compressed = sp.append(0, sp.cumsum(sp.bincount(idx_centroids, minlength=self.request.npos)))
backward_idx = idx_req[sp.argsort(idx_centroids)]
self.backward_idx = (cuda.to_gpu_correct(backward_compressed),
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment