Commit 55051166 authored by NaOHCC's avatar NaOHCC Committed by LeiWang1999
Browse files

[Bugfix] Correct method call for block reduction check when analyzing memory footprint (#299)

parent 3f294650
...@@ -493,7 +493,7 @@ class PrimFuncNode(Node): ...@@ -493,7 +493,7 @@ class PrimFuncNode(Node):
for buffer in self.block_analyzer.get_input_buffers(block): for buffer in self.block_analyzer.get_input_buffers(block):
cache = buffer.name not in cached_tensor and ( cache = buffer.name not in cached_tensor and (
is_broadcast_pattern(buffer, output_buffer) or is_broadcast_pattern(buffer, output_buffer) or
self.block_analyzer.get_block_info(block).is_reduction) self.block_analyzer.get_block_info(block).is_reduction())
if not cache: if not cache:
continue continue
cached_tensor.append(buffer.name) cached_tensor.append(buffer.name)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment