"vscode:/vscode.git/clone" did not exist on "5fe29b063f7877036e54e26d85c21e2d1d51a2c8"
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):
for buffer in self.block_analyzer.get_input_buffers(block):
cache = buffer.name not in cached_tensor and (
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:
continue
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