Module Expr Str ImportFrom alias Import alias Import alias identifier Import alias identifier Import alias identifier Expr Call NameLoad Str Assign NameStore Num Assign NameStore BinOpFloorDiv NameLoad Num Assign NameStore Call AttributeLoad NameLoad attr Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad NameLoad AttributeLoad NameLoad attr Assign NameStore Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad Assign NameStore Call AttributeLoad NameLoad attr Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad Assign NameStore Call AttributeLoad NameLoad attr NameLoad Assign NameStore Call AttributeLoad NameLoad attr TupleLoad SubscriptLoad AttributeLoad NameLoad attr Index Num BinOpAdd BinOpFloorDiv SubscriptLoad AttributeLoad NameLoad attr Index Num Num Num AttributeLoad NameLoad attr Assign NameStore Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr AttributeLoad NameLoad attr AttributeLoad NameLoad attr Expr Call AttributeLoad NameLoad attr NameLoad NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr NameLoad Assign NameStore Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr AttributeLoad NameLoad attr AttributeLoad NameLoad attr Expr Call AttributeLoad NameLoad attr NameLoad NameLoad NameLoad NameLoad Expr Call NameLoad Str Call AttributeLoad NameLoad attr NameLoad Call AttributeLoad NameLoad attr keyword Num Expr Call NameLoad Str Assign NameStore Call AttributeLoad NameLoad attr BinOpAdd Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad NameLoad BinOpMult Num Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad NameLoad AttributeLoad NameLoad attr Assign NameStore Call AttributeLoad NameLoad attr NameLoad Assign NameStore Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr AttributeLoad NameLoad attr AttributeLoad NameLoad attr Expr Call AttributeLoad NameLoad attr NameLoad NameLoad NameLoad Expr Call AttributeLoad NameLoad attr NameLoad NameLoad NameLoad NameLoad Expr Call NameLoad Str Call AttributeLoad NameLoad attr NameLoad Call AttributeLoad NameLoad attr keyword Num
N/A N/A 
Demonstrates how to use the PyCUDA interface to CUFFT to compute 2D FFTs.
 __future__ print_function N/A pycuda.autoinit N/A pycuda.gpuarray gpuarray N/A numpy np N/A skcuda.fft cu_fft N/A N/A print Testing fft/ifft.. N/A N 1024 N/A M N/A N 2 N/A x N/A N/A np asarray N/A N/A N/A np random rand N M N/A np float32 N/A xf N/A N/A N/A np fft fft2 x N/A y N/A N/A np real N/A N/A N/A np fft ifft2 xf N/A x_gpu N/A N/A gpuarray to_gpu x N/A xf_gpu N/A N/A gpuarray empty N/A N/A N/A x shape N/A 0 N/A N/A N/A N/A x shape N/A 1 2 1 N/A np complex64 N/A plan_forward N/A N/A cu_fft Plan N/A x_gpu shape N/A np float32 N/A np complex64 N/A N/A N/A cu_fft fft x_gpu xf_gpu plan_forward N/A y_gpu N/A N/A gpuarray empty_like x_gpu N/A plan_inverse N/A N/A cu_fft Plan N/A x_gpu shape N/A np complex64 N/A np float32 N/A N/A N/A cu_fft ifft xf_gpu y_gpu plan_inverse True N/A N/A print Success status:  N/A N/A np allclose y N/A N/A y_gpu get atol 1e-06 N/A N/A print Testing in-place fft.. N/A x N/A N/A np asarray N/A N/A N/A N/A np random rand N M N/A 1j N/A N/A N/A np random rand N M N/A np complex64 N/A x_gpu N/A N/A gpuarray to_gpu x N/A plan N/A N/A cu_fft Plan N/A x_gpu shape N/A np complex64 N/A np complex64 N/A N/A N/A cu_fft fft x_gpu x_gpu plan N/A N/A N/A cu_fft ifft x_gpu x_gpu plan True N/A N/A print Success status:  N/A N/A np allclose x N/A N/A x_gpu get atol 1e-06