From e436ed7b22cc462a294d84ef808d2e6b00287adb Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Mon, 30 Aug 2021 11:14:18 -0500
Subject: [PATCH] WalkMapper: fix map_einsum (#147)

* WalkMapper: fix map_einsum

* Update transform.py
---
 pytato/transform.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pytato/transform.py b/pytato/transform.py
index 2e252df..b419265 100644
--- a/pytato/transform.py
+++ b/pytato/transform.py
@@ -500,6 +500,8 @@ class WalkMapper(Mapper):
             if isinstance(dim, Array):
                 self.rec(dim)
 
+        self.post_visit(expr)
+
     def map_loopy_call(self, expr: LoopyCall) -> None:
         if not self.visit(expr):
             return
-- 
GitLab