Commit cf9b8764 authored by Michiel Cottaar's avatar Michiel Cottaar
Browse files

BUG: correctly extract radial results from MultEvaluator

parent 89d3828a
Pipeline #5300 failed with stage
in 8 minutes and 34 seconds
......@@ -409,6 +409,10 @@ class _RadialGPUArrays(object):
arr.fill(0.)
self.is_set = set()
def delete(self, component):
if component in self.arrays:
del self.arrays[component]
def __contains__(self, item):
return item in self.arrays
......@@ -670,9 +674,9 @@ __global__ void matrix_mult_nd_invert({dtype} *derparam, {dtype} *derfield, int
def clean_results(self, ):
RadialGPUArrays.clean()
def update_pos(self, new_positions):
def update_pos(self, new_request):
RadialGPUArrays.set(self.request, RadialGPUArrays.get(self.request)[:new_request.npos])
idx = self.req_params_cuda_names.index('all_pos')
self.global_cuda_params[self.request][idx][:new_positions.size] = new_positions.astype(cuda.dtype).flatten()
self.request.positions[:new_positions.shape[0]] = new_positions
self.global_cuda_params[self.request][idx][:new_request.positions.size] = new_request.positions.astype(cuda.dtype).flatten()
self.update_indices(only_forward=True)
Markdown is supported
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