no consumer map

Signed-off-by: Masaki Kozuki <mkozuki@nvidia.com>
Lightning-AI · Dec 16, 2024 · 29c20ca · 29c20ca
1 parent c026ee4
commit 29c20ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/thunder/transforms/tensor_subclasses.py b/thunder/transforms/tensor_subclasses.py
@@ -646,7 +646,7 @@ def tensor_subclass_dce(trace: TraceCtx) -> TraceCtx:
     """
     start_time_ns = time.perf_counter_ns()
     swap_map: dict[Variable, TensorProxy] = {}
-    producer_map, consumer_map = utils.producers_and_consumers(trace)
+    producer_map = utils.producers(trace)
     bsym_to_exclude: set[BoundSymbol] = set()
 
     subclass_flatten_bsym: BoundSymbol