import numpy as np import pycuda.autoinit from nervanagpu import NervanaGPU nrv = NervanaGPU(default_dtype=np.float32) a = nrv.array(np.random.randn(200,200)) b = nrv.empty_like(a) b[:] = a**2 assert not np.any(np.isnan(b.get())), "Shouldn't have any nan's here"