From 902036357c97ed98219d0ccca7304d4811e02805 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Sun, 20 Dec 2020 13:18:58 -0600 Subject: [PATCH] Drop usage of six --- test/undistributed/elwise-perf.py | 5 ----- test/undistributed/measure_gpuarray_speed.py | 5 ----- test/undistributed/reduction-perf.py | 8 +------- 3 files changed, 1 insertion(+), 17 deletions(-) diff --git a/test/undistributed/elwise-perf.py b/test/undistributed/elwise-perf.py index dc2c2821..8562b99c 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 26a20f41..4f6f390e 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 842824cd..da4a0d03 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) -- GitLab