func (c *DemagConvolution) init() {
{ // init FFT plans
padded := c.kernSize
c.stream = cu.StreamCreate()
c.fwPlan = newFFT3DR2C(padded[0], padded[1], padded[2], c.stream)
c.bwPlan = newFFT3DC2R(padded[0], padded[1], padded[2], c.stream)
}
{ // init device buffers
// 2D re-uses fftBuf[1] as fftBuf[0], 3D needs all 3 fftBufs.
for i := 1; i < 3; i++ {
c.fftCBuf[i] = makeFloats(fftR2COutputSizeFloats(c.kernSize))
}
if c.is3D() {
c.fftCBuf[0] = makeFloats(fftR2COutputSizeFloats(c.kernSize))
} else {
c.fftCBuf[0] = c.fftCBuf[1]
}
for i := 0; i < 3; i++ {
c.fftRBuf[i] = c.fftCBuf[i].Slice(0, prod(c.kernSize))
}
}
if c.is2D() {
c.initFFTKern2D()
} else {
c.initFFTKern3D()
}
}
func initStreampool() {
streamPool = make(chan cu.Stream, streamPoolSize)
for i := 0; i < streamPoolSize; i++ {
streamPool <- cu.StreamCreate()
}
}