"docs/zh_cn/git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "35affc35bff30848c89db937f29e68c0da496ec5"
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