file theano/misc/tests/test_pycuda_theano_simple.py

theano/misc/tests/test_pycuda_theano_simple.py

@@ -8,7 +8,7 @@ from __future__ import absolute_import, print_function, division
 
 import sys
 
-import numpy
+import numpy as np
 
 import theano
 import theano.sandbox.cuda as cuda_ndarray
@@ -42,9 +42,9 @@ __global__ void multiply_them(float *dest, float *a, float *b)
     multiply_them = mod.get_function("multiply_them")
 
     # Test with pycuda in/out of numpy.ndarray
-    a = numpy.random.randn(100).astype(numpy.float32)
-    b = numpy.random.randn(100).astype(numpy.float32)
-    dest = numpy.zeros_like(a)
+    a = np.random.randn(100).astype(np.float32)
+    b = np.random.randn(100).astype(np.float32)
+    dest = np.zeros_like(a)
     multiply_them(
         drv.Out(dest), drv.In(a), drv.In(b),
         block=(400, 1, 1), grid=(1, 1))
@@ -64,8 +64,8 @@ __global__ void multiply_them(float *dest, float *a, float *b)
 
     multiply_them = mod.get_function("multiply_them")
 
-    a = numpy.random.randn(100).astype(numpy.float32)
-    b = numpy.random.randn(100).astype(numpy.float32)
+    a = np.random.randn(100).astype(np.float32)
+    b = np.random.randn(100).astype(np.float32)
 
     # Test with Theano object
     ga = cuda_ndarray.CudaNdarray(a)
@@ -73,7 +73,7 @@ __global__ void multiply_them(float *dest, float *a, float *b)
     dest = cuda_ndarray.CudaNdarray.zeros(a.shape)
     multiply_them(dest, ga, gb,
                   block=(400, 1, 1), grid=(1, 1))
-    assert (numpy.asarray(dest) == a * b).all()
+    assert (np.asarray(dest) == a * b).all()
 
 
 def test_pycuda_memory_to_theano():
@@ -87,7 +87,7 @@ def test_pycuda_memory_to_theano():
     print("gpuarray ref count before creating a CudaNdarray", end=' ')
     print(sys.getrefcount(y))
     assert sys.getrefcount(y) == initial_refcount
-    rand = numpy.random.randn(*y.shape).astype(numpy.float32)
+    rand = np.random.randn(*y.shape).astype(np.float32)
     cuda_rand = cuda_ndarray.CudaNdarray(rand)
 
     strides = [1]
@@ -102,7 +102,7 @@ def test_pycuda_memory_to_theano():
     z = cuda_ndarray.from_gpu_pointer(y_ptr, y.shape, strides, y)
     print("gpuarray ref count after creating a CudaNdarray", sys.getrefcount(y))
     assert sys.getrefcount(y) == initial_refcount + 1
-    assert (numpy.asarray(z) == 0).all()
+    assert (np.asarray(z) == 0).all()
     assert z.base is y
 
     # Test that we can take a view from this cuda view on pycuda memory
@@ -112,17 +112,17 @@ def test_pycuda_memory_to_theano():
     del zz
     assert sys.getrefcount(y) == initial_refcount + 1
 
-    cuda_ones = cuda_ndarray.CudaNdarray(numpy.asarray([[[1]]],
-                                                       dtype='float32'))
+    cuda_ones = cuda_ndarray.CudaNdarray(np.asarray([[[1]]],
+                                                    dtype='float32'))
     z += cuda_ones
-    assert (numpy.asarray(z) == numpy.ones(y.shape)).all()
-    assert (numpy.asarray(z) == 1).all()
+    assert (np.asarray(z) == np.ones(y.shape)).all()
+    assert (np.asarray(z) == 1).all()
 
     assert cuda_rand.shape == z.shape
     assert cuda_rand._strides == z._strides, (cuda_rand._strides, z._strides)
-    assert (numpy.asarray(cuda_rand) == rand).all()
+    assert (np.asarray(cuda_rand) == rand).all()
     z += cuda_rand
-    assert (numpy.asarray(z) == (rand + 1)).all()
+    assert (np.asarray(z) == (rand + 1)).all()
 
     # Check that the ref count to the gpuarray is right.
     del z