Commit e51e11bf authored by Pierre Paleo's avatar Pierre Paleo
Browse files

Assume delta_z is a multiple of chunk_size (2)

parent 8ff03259
Pipeline #26894 passed with stages
in 3 minutes and 1 second
......@@ -250,15 +250,15 @@ class CudaFullFieldPipelineLimitedMemory(CudaFullFieldPipeline):
self.logger.debug("processing stack %d/%d" % (i+1, n_groups))
start_idx = i * group_size
end_idx = min((i + 1) * group_size, self.delta_z)
transfer_size = end_idx - start_idx
transfer_size = end_idx - start_idx # not useful here as delta_z % chunk_size == 0
# Build sinograms on host. Use a view of _h_radios
self._build_sino(radios=self._h_radios[:, start_idx:end_idx, :])
# Copy H2D
self._d_sinos[:transfer_size, :, :] = self._sinobuilder_output[:transfer_size, :, :]
self._d_sinos[:, :, :] = self._sinobuilder_output[:, :, :]
# Process stack of sinograms (chunk_size, n_angles, width)
self._reconstruct()
# Copy D2H
self._d_recs[:transfer_size, :, :].get(ary=self._h_recs[start_idx:end_idx])
self._d_recs.get(ary=self._h_recs[start_idx:end_idx])
self.logger.debug("End of processing steps on sinos")
# Write
self._write_data(data=self._h_recs)
......
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