Skip to content
Snippets Groups Projects
Commit 8c3b6164 authored by René Heß's avatar René Heß
Browse files

Use Mapper from loopy and rewiting handlers

parent 77c56f64
No related branches found
No related tags found
No related merge requests found
......@@ -41,7 +41,7 @@ from pytools import ImmutableRecord
import loopy as lp
import numpy as np
import pymbolic.primitives as prim
from pymbolic.mapper import WalkMapper
from loopy.symbolic import WalkMapper
import ufl.classes as uc
from ufl import FiniteElement, MixedElement, TensorProductElement
......@@ -50,14 +50,6 @@ basis_sf_kernels = generator_factory(item_tags=("basis_sf_kernels",), context_ta
class SumfactCollectMapper(WalkMapper):
def map_tagged_variable(self, expr, *args, **kwargs):
self.visit(expr, *args, **kwargs)
self.post_visit(expr, *args, **kwargs)
def map_variable(self, expr, *args, **kwargs):
self.visit(expr, *args, **kwargs)
self.post_visit(expr, *args, **kwargs)
def map_sumfact_kernel(self, expr, *args, **kwargs):
basis_sf_kernels(expr)
self.visit(expr, *args, **kwargs)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment