diff --git a/test/undistributed/elwise-perf.py b/test/undistributed/elwise-perf.py index dc2c28211559bf00e316d6a1f6536512e73a17e3..8562b99c0d5afb3296a628c3ee2d9e0d18c3514d 100644 --- a/test/undistributed/elwise-perf.py +++ b/test/undistributed/elwise-perf.py @@ -1,12 +1,7 @@ -from __future__ import absolute_import -from __future__ import print_function - -#! /usr/bin/env python import pycuda.driver as drv import pycuda.autoinit import numpy import numpy.linalg as la -from six.moves import range def main(): diff --git a/test/undistributed/measure_gpuarray_speed.py b/test/undistributed/measure_gpuarray_speed.py index 26a20f41b65ac3b6b7da4eb1bc13612b261db2af..4f6f390e621aa90cbcd9f19f7eacab44e6773c11 100755 --- a/test/undistributed/measure_gpuarray_speed.py +++ b/test/undistributed/measure_gpuarray_speed.py @@ -1,13 +1,8 @@ -from __future__ import absolute_import -from __future__ import print_function - #! /usr/bin/env python import pycuda.driver as drv import pycuda.autoinit import numpy import numpy.linalg as la -from six.moves import range -from six.moves import zip def main(): diff --git a/test/undistributed/reduction-perf.py b/test/undistributed/reduction-perf.py index 842824cdb463e6d8c669aa1fd386259eed862440..da4a0d03c8a62e78838a8acb202dd9f452ffcf08 100644 --- a/test/undistributed/reduction-perf.py +++ b/test/undistributed/reduction-perf.py @@ -1,11 +1,7 @@ -from __future__ import division -from __future__ import absolute_import -from __future__ import print_function import pycuda.autoinit import pycuda.gpuarray as gpuarray import pycuda.driver as cuda import numpy -from six.moves import range def main(): @@ -14,8 +10,6 @@ def main(): tbl = Table() tbl.add_row(("type", "size [MiB]", "time [ms]", "mem.bw [GB/s]")) - from random import shuffle - for dtype_out in [numpy.float32, numpy.float64]: for ex in range(15, 27): sz = 1 << ex @@ -28,7 +22,7 @@ def main(): assert sz == a_gpu.shape[0] assert len(a_gpu.shape) == 1 - from pycuda.reduction import get_sum_kernel, get_dot_kernel + from pycuda.reduction import get_dot_kernel krnl = get_dot_kernel(dtype_out, a_gpu.dtype)