[哈工大] tf.image.extract_image_patches算子出错

代码如下,其中mask_pad的shape为(16, 1, 319, 319, 16)，g_sz=127
```
    mask_uf = tf.image.extract_image_patches(mask_pad, ksizes=[1, g_sz, g_sz, 1],
                                             strides=[1, 8, 8, 1],
                                             rates=[1, 1, 1, 1],
                                             padding='VALID')
```
报错信息如下：
```
[ERROR] TEFUSION(194884,python3.7):2020-12-10-15:11:56.811.220 [tensor_engine/te_fusion/fusion_op.cc:5713]GetFinishedCompilationTask FinishedTask[0]: taskID[281446916809200:1637], status[1], kernel[None] res: prebuild failed. module[impl.extract_image_patches] func[extract_image_patches], compile_info: 0
[ERROR] TEFUSION(194884,python3.7):2020-12-10-15:11:56.811.334 [tensor_engine/te_fusion/fusion_op.cc:5716]GetFinishedCompilationTask Op error args: args:(), input:({'shape': (16, 1, 319, 319, 16), 'ori_shape': (16, 319, 319, 1), 'format': 'NC1HWC0', 'ori_format': 'NHWC', 'dtype': 'float16', 'addr_type': 0, 'valid_shape': (), 'slice_offset': (), 'L1_workspace_size': -1, 'L1_fusion_type': -1, 'L1_addr_offset': 0, 'total_shape': (), 'split_index': 0},), outputs:({'shape': (16, 25, 25, 16129), 'ori_shape': (16, 25, 25, 16129), 'format': 'NHWC', 'ori_format': 'NHWC', 'dtype': 'float16', 'addr_type': 0, 'valid_shape': (), 'slice_offset': (), 'L1_workspace_size': -1, 'L1_fusion_type': -1, 'L1_addr_offset': 0, 'total_shape': (), 'split_index': 0},), attrs:((1, 127, 127, 1), (1, 8, 8, 1), (1, 1, 1, 1), 'VALID')
[ERROR] TEFUSION(194884,python3.7):2020-12-10-15:11:56.811.369 [tensor_engine/te_fusion/fusion_op.cc:5720]GetFinishedCompilationTask Op python exception: Traceback (most recent call last):
[ERROR] TEFUSION(194884,python3.7):2020-12-10-15:11:56.811.390 [tensor_engine/te_fusion/fusion_op.cc:5720]GetFinishedCompilationTask   File "/usr/local/lib/python3.7/site-packages/te/platform/parallel_compilation.py", line 1184, in run
[ERROR] TEFUSION(194884,python3.7):2020-12-10-15:11:56.811.408 [tensor_engine/te_fusion/fusion_op.cc:5720]GetFinishedCompilationTask     int64_mode=self._int64_mode)
[ERROR] TEFUSION(194884,python3.7):2020-12-10-15:11:56.811.426 [tensor_engine/te_fusion/fusion_op.cc:5720]GetFinishedCompilationTask   File "/usr/local/lib/python3.7/site-packages/te/platform/fusion_manager.py", line 866, in build_single_op
[ERROR] TEFUSION(194884,python3.7):2020-12-10-15:11:56.811.443 [tensor_engine/te_fusion/fusion_op.cc:5720]GetFinishedCompilationTask     compile_info = call_op()
[ERROR] TEFUSION(194884,python3.7):2020-12-10-15:11:56.811.460 [tensor_engine/te_fusion/fusion_op.cc:5720]GetFinishedCompilationTask   File "/usr/local/lib/python3.7/site-packages/te/platform/fusion_manager.py", line 851, in call_op
[ERROR] TEFUSION(194884,python3.7):2020-12-10-15:11:56.811.478 [tensor_engine/te_fusion/fusion_op.cc:5720]GetFinishedCompilationTask     opfunc(*inputs, *outputs, *attrs, **kwargs)
[ERROR] TEFUSION(194884,python3.7):2020-12-10-15:11:56.811.496 [tensor_engine/te_fusion/fusion_op.cc:5720]GetFinishedCompilationTask   File "/usr/local/lib/python3.7/site-packages/te/utils/op_utils.py", line 597, in _in_wrapper
[ERROR] TEFUSION(194884,python3.7):2020-12-10-15:11:56.811.513 [tensor_engine/te_fusion/fusion_op.cc:5720]GetFinishedCompilationTask     return func(*args, **kwargs)
[ERROR] TEFUSION(194884,python3.7):2020-12-10-15:11:56.811.530 [tensor_engine/te_fusion/fusion_op.cc:5720]GetFinishedCompilationTask   File "/home/HwHiAiUser/Ascend/ascend-toolkit/latest/arm64-linux/opp/op_impl/built-in/ai_core/tbe/impl/extract_image_patches.py", line 876, in extract_image_patches
[ERROR] TEFUSION(194884,python3.7):2020-12-10-15:11:56.811.548 [tensor_engine/te_fusion/fusion_op.cc:5720]GetFinishedCompilationTask     (cut_h_col * fmap_w * fmap_c0 * type_size * DOUBLE_BUFFER))
[ERROR] TEFUSION(194884,python3.7):2020-12-10-15:11:56.811.565 [tensor_engine/te_fusion/fusion_op.cc:5720]GetFinishedCompilationTask RuntimeError: Input size is too large load to L1, while cut h, need size: 3756544
[ERROR] FE(194884,python3.7):2020-12-10-15:11:56.821.805 [fusion_engine/adapter/tbe_adapter/tbe_op_store_adapter.cpp:135]196108 ProcessFailPreCompTask:"tid[281446916809200], taskId[1637], node[model/ExtractImagePatches], precompile failed"
```

Ascend/modelzoo
暂停

内容风险标识

评论 (15)

Ascend/modelzoo暂停 .gitee-modal { width: 500px !important; }

内容风险标识