func ExampleFFT1D() {
N := 8
hostIn := make([]float32, N)
hostIn[0] = 1
devIn := cu.MemAlloc(int64(len(hostIn)) * cu.SIZEOF_FLOAT32)
defer cu.MemFree(&devIn)
cu.MemcpyHtoD(devIn, unsafe.Pointer(&hostIn[0]), devIn.Bytes())
hostOut := make([]complex64, N/2+1)
devOut := cu.MemAlloc(int64(len(hostOut)) * cu.SIZEOF_COMPLEX64)
defer cu.MemFree(&devOut)
plan := Plan1d(N, R2C, 1)
defer plan.Destroy()
plan.ExecR2C(devIn, devOut)
cu.MemcpyDtoH(unsafe.Pointer(&hostOut[0]), devOut, devOut.Bytes())
fmt.Println("hostIn:", hostIn)
fmt.Println("hostOut:", hostOut)
// Output:
// hostIn: [1 0 0 0 0 0 0 0]
// hostOut: [(1+0i) (+1+0i) (+1+0i) (+1-0i) (+1+0i)]
}
func (src *slice) copyDtoH(dst unsafe.Pointer, dstlen int, elemsize int) {
if dstlen != src.Len() {
panic(fmt.Errorf("cuda4/safe: len mismatch: src.Len()=%v (device), len(dst)=%v (host)", src.Len(), dstlen))
}
cu.MemcpyDtoH(dst, src.Pointer(), int64(elemsize)*int64(dstlen))
}