From cb659cca124e2e667150cf61585c3ed458716e8f Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 3 May 2015 23:57:34 -0500
Subject: [PATCH] Make loopy pymbolic primitives accessible in loopy namespace
---
loopy/__init__.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/loopy/__init__.py b/loopy/__init__.py
index 2bea55f08..db6953a6d 100644
--- a/loopy/__init__.py
+++ b/loopy/__init__.py
@@ -31,7 +31,8 @@ import islpy as isl
from islpy import dim_type
from loopy.symbolic import (RuleAwareIdentityMapper, RuleAwareSubstitutionMapper,
- SubstitutionRuleMappingContext)
+ SubstitutionRuleMappingContext,
+ TaggedVariable, Reduction, LinearSubscript, )
from loopy.diagnostic import LoopyError
@@ -62,12 +63,14 @@ from loopy.padding import (split_arg_axis, find_padding_multiple,
from loopy.preprocess import (preprocess_kernel, realize_reduction,
infer_unknown_types)
from loopy.schedule import generate_loop_schedules, get_one_scheduled_kernel
-from loopy.codegen import generate_code
+from loopy.codegen import generate_code, generate_body
from loopy.compiled import CompiledKernel
from loopy.options import Options
from loopy.auto_test import auto_test_vs_ref
__all__ = [
+ "TaggedVariable", "Reduction", "LinearSubscript",
+
"auto",
"LoopKernel",
--
GitLab