def test_cublasDrnm2(self):
x = np.random.rand(5).astype(np.float64)
x_gpu = gpuarray.to_gpu(x)
result = cublas.cublasDnrm2(self.cublas_handle, x_gpu.size, x_gpu.gpudata, 1)
assert np.allclose(result, np.linalg.norm(x))
def _l2norm(self, handle, x_gpu):
return cublas.cublasDnrm2(self.h, x_gpu.size, x_gpu.gpudata, 1)