From 300c6960d2cd52509ead7df88b93aff336ef3907 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Thu, 27 Nov 2014 15:21:03 -0600 Subject: [PATCH] More Py3 compatibility --- loopy/preprocess.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/loopy/preprocess.py b/loopy/preprocess.py index 04e550a87..06268a290 100644 --- a/loopy/preprocess.py +++ b/loopy/preprocess.py @@ -1,8 +1,4 @@ -from __future__ import division -from __future__ import absolute_import -import six -from six.moves import range -from six.moves import zip +from __future__ import division, absolute_import __copyright__ = "Copyright (C) 2012 Andreas Kloeckner" @@ -27,6 +23,9 @@ THE SOFTWARE. """ +import six +from six.moves import range, zip +import numpy as np import pyopencl as cl import pyopencl.characterize as cl_char from loopy.diagnostic import ( @@ -836,9 +835,10 @@ def get_auto_axis_iname_ranking_by_stride(kernel, insn): aggregate_strides[iname] = aggregate_strides.get(iname, 0) + stride if aggregate_strides: - import sys + very_large_stride = np.iinfo(np.int32).max + return sorted((iname for iname in kernel.insn_inames(insn)), - key=lambda iname: aggregate_strides.get(iname, sys.maxint)) + key=lambda iname: aggregate_strides.get(iname, very_large_stride)) else: return None -- GitLab