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

CudaSinoNormalization: fix grid and block

parent 37489777
Pipeline #33315 passed with stages
in 2 minutes and 23 seconds
......@@ -111,12 +111,12 @@ class CudaSinoNormalization(SinoNormalization, CudaProcessing):
signature="Piii",
)
self._cuda_kernel_args = [
np.int32(self.n_z), np.int32(self.n_angles), np.int32(self.n_x)
np.int32(self.n_x), np.int32(self.n_angles), np.int32(self.n_z)
]
blk = (64, 1, 16) # TODO tune ?
blk = (1, 64, 16) # TODO tune ?
self._cuda_kernel_kwargs = {
"block": blk,
"grid": (updiv(self.n_z, blk[0]), 1, updiv(self.n_x, blk[2])),
"grid": (1, updiv(self.n_angles, blk[1]), updiv(self.n_z, blk[2])),
}
......
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