import cupy as cp

dtype = cp.complex128
shape = (200, 1000, 1000)
x = cp.random.random(shape).astype(dtype) \
    +1j*cp.random.random(shape).astype(dtype)

stream = cp.cuda.Stream()
with stream:
    y = x.sum()  # runs on "stream" instead of the default stream