import numpy as np
import cupy as cp

print("CuPy:", cp.__version__)
print("GPU:", cp.cuda.runtime.getDeviceProperties(0)["name"])

x_cpu = np.random.random(1024 * 1024).astype(np.complex64)
x_gpu = cp.asarray(x_cpu)

y_gpu = cp.fft.fft(x_gpu)
cp.cuda.Stream.null.synchronize()

print(y_gpu[:5])
print("cuFFT OK")