nabu.processing.muladd_cuda
source module nabu.processing.muladd_cuda
Classes
source class CudaMulAdd(**backend_options)
Bases : MulAdd
Methods
-
mul_add — Performs dst[DST_IDX] = fac_dstdst[DST_IDX] + fac_otherother[OTHER_IDX] where DST_IDX = dst_start_row:dst_end_row, dst_start_col:dst_end_col OTHER_IDX = other_start_row:other_end_row, other_start_col:other_end_col
source method CudaMulAdd.mul_add(dst, other, fac_dst, fac_other, dst_region=None, other_region=None)
Performs dst[DST_IDX] = fac_dstdst[DST_IDX] + fac_otherother[OTHER_IDX] where DST_IDX = dst_start_row:dst_end_row, dst_start_col:dst_end_col OTHER_IDX = other_start_row:other_end_row, other_start_col:other_end_col
'region' should be a tuple (slice(y_start, y_end), slice(x_start, x_end))
Raises
-
ValueError