From da8d7c8e1ff403c1f1553b240d8a3546d3b767b5 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Mon, 7 Nov 2011 11:40:42 -0500 Subject: [PATCH] Some code rearrangement. --- loopy/codegen/__init__.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/loopy/codegen/__init__.py b/loopy/codegen/__init__.py index fb3e00bd3..a4dbafe64 100644 --- a/loopy/codegen/__init__.py +++ b/loopy/codegen/__init__.py @@ -183,10 +183,11 @@ def generate_code(kernel): POD, Value, ArrayOf, Module, Block, Line, Const, LiteralLines, Initializer) - from cgen.opencl import (CLKernel, CLGlobal, CLConstant, CLRequiredWorkGroupSize, + from cgen.opencl import (CLKernel, CLGlobal, CLRequiredWorkGroupSize, CLLocal, CLImage, CLConstant) - from loopy.symbolic import LoopyCCodeMapper + from loopy.symbolic import LoopyCCodeMapper, pw_aff_to_expr + ccm = LoopyCCodeMapper(kernel).copy_and_assign_many( make_initial_assignments(kernel)) @@ -286,7 +287,6 @@ def generate_code(kernel): except AttributeError: storage_shape = tv.shape - from loopy.symbolic import pw_aff_to_expr for l in storage_shape: temp_var_decl = ArrayOf(temp_var_decl, l) @@ -311,7 +311,6 @@ def generate_code(kernel): else: body.append(gen_code.ast) - from loopy.symbolic import pw_aff_to_expr mod.append( FunctionBody( CLRequiredWorkGroupSize( -- GitLab