From a3f90b2959e463cfda5aee952277bd1c9bbd5a4f Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Fri, 3 Jun 2016 17:27:27 -0500 Subject: [PATCH] Fix some pymbolic imports --- loopy/transform/diff.py | 5 ++--- loopy/transform/subst.py | 2 +- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/loopy/transform/diff.py b/loopy/transform/diff.py index 0d80ee99c..1f75a60b8 100644 --- a/loopy/transform/diff.py +++ b/loopy/transform/diff.py @@ -296,10 +296,9 @@ class DifferentiationContext(object): return None assert isinstance(orig_writer_insn, lp.Assignment) - from pymbolic import Variable, Subscript - if isinstance(orig_writer_insn.assignee, Subscript): + if isinstance(orig_writer_insn.assignee, p.Subscript): lhs_ind = orig_writer_insn.assignee.index_tuple - elif isinstance(orig_writer_insn.assignee, Variable): + elif isinstance(orig_writer_insn.assignee, p.Variable): lhs_ind = () else: raise LoopyError( diff --git a/loopy/transform/subst.py b/loopy/transform/subst.py index 6cbf43da3..27e98383e 100644 --- a/loopy/transform/subst.py +++ b/loopy/transform/subst.py @@ -394,7 +394,7 @@ def assignment_to_subst(kernel, lhs_name, extra_arguments=(), within=None, from loopy.kernel.data import Assignment assert isinstance(def_insn, Assignment) - from pymbolic import Variable, Subscript + from pymbolic.primitives import Variable, Subscript if isinstance(def_insn.assignee, Subscript): indices = def_insn.assignee.index_tuple elif isinstance(def_insn.assignee, Variable): -- GitLab