File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 5353 cmp (dZ, X)
5454
5555 # region/batched (1D along dim 1)
56- X = rand (T, Ns[1 ], Ns[2 ])
57- dX = gpu (X)
58- p = plan_fft! (dX, 1 )
59- p * dX
60- cmp (dX, fft (X,1 ))
61- pinv = plan_ifft! (dX,1 )
62- pinv * dX
63- cmp (dX, X)
56+ # X = rand(T, Ns[1], Ns[2])
57+ # dX = gpu(X)
58+ # p = plan_fft!(dX, 1)
59+ # p * dX
60+ # cmp(dX, fft(X,1))
61+ # pinv = plan_ifft!(dX,1)
62+ # pinv * dX
63+ # cmp(dX, X)
6464 end
6565 end
6666
7777 cmp (dZ, X)
7878
7979 # 2D real rfft along first dim default
80- X = rand (T, Ns[1 ], Ns[2 ])
81- dX = gpu (X)
82- p = plan_rfft (dX)
83- dY = p * dX
84- cmp (dY, rfft (X))
85- pinv = plan_irfft (dY, size (X,1 ))
86- dZ = pinv * dY
87- cmp (dZ, X)
80+ # X = rand(T, Ns[1], Ns[2])
81+ # dX = gpu(X)
82+ # p = plan_rfft(dX)
83+ # dY = p * dX
84+ # cmp(dY, rfft(X))
85+ # pinv = plan_irfft(dY, size(X,1))
86+ # dZ = pinv * dY
87+ # cmp(dZ, X)
8888 end
8989 end
9090
You can’t perform that action at this time.
0 commit comments