no lambdas

Lightning-AI · Dec 12, 2024 · 1969d9a · 1969d9a
1 parent fe66455
commit 1969d9a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/thunder/executors/torch_compile.py b/thunder/executors/torch_compile.py
@@ -86,7 +86,9 @@ def make_compiled(
             if o is not None:
                 region_trace.add_name(o.name)
         for sbsym in bsym.subsymbols:
-            list(map(lambda o: region_trace.add_name(o.name), filter(lambda o: o is not None and o.name not in region_trace.names, sbsym.flat_outs)))
+            for o in sbsym.flat_outs:
+                if o is not None and o.name not in region_trace.names:
+                    region_trace.add_name(o.name)
 
     # maybe make this the default if no sig info is present?
     region_trace._siginfo = SigInfo("to_be_compiled")