Skip to content
Snippets Groups Projects
Commit add4ee09 authored by Matt Wala's avatar Matt Wala
Browse files

Flake8 fixes.

parent 4f9b02ae
Branches
Tags
1 merge request!23Faster CSE
Pipeline #
......@@ -76,23 +76,6 @@ class E2EBase(KernelCacheWrapper):
self.dim = src_expansion.dim
def run_translation_and_cse(self):
from sumpy.symbolic import make_sympy_vector
dvec = make_sympy_vector("d", self.dim)
src_coeff_exprs = [sp.Symbol("src_coeff%d" % i)
for i in range(len(self.src_expansion))]
from sumpy.assignment_collection import SymbolicAssignmentCollection
sac = SymbolicAssignmentCollection()
tgt_coeff_names = [
sac.assign_unique("coeff%d" % i, coeff_i)
for i, coeff_i in enumerate(
self.tgt_expansion.translate_from(
self.src_expansion, src_coeff_exprs, dvec))]
sac.run_global_cse()
def get_translation_loopy_insns(self):
from sumpy.symbolic import make_sympy_vector
dvec = make_sympy_vector("d", self.dim)
......
......@@ -64,18 +64,14 @@ DAMAGE.
# }}}
import pytest
import itertools
import sys
from sympy import (Add, Pow, Symbol, exp, sqrt, symbols, sympify, S, cos,
sin, Eq, Function, Tuple, CRootOf, IndexedBase, Idx,
Piecewise)
from sympy import (Add, Pow, exp, sqrt, symbols, sympify, S, cos, sin, Eq,
Function, Tuple, CRootOf, IndexedBase, Idx, Piecewise)
from sympy.simplify.cse_opts import sub_pre, sub_post
from sympy.functions.special.hyper import meijerg
from sympy.simplify import cse_opts
from sympy.core.compatibility import range
from sumpy.cse import (
cse, preprocess_for_cse, postprocess_for_cse)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment