elementwise_mul算子无法在METAL环境正常使用
-
版本、环境信息: 1)Paddle Lite 版本:2.14.rc 2)Host 环境:M4处理器,MacOS15,python3.10
-
模型信息 1)模型名称:PaddleOCR v4
-
复现信息:
-
模型转换命令paddle_lite_opt --model_dir=./ch_rec_fast --optimize_out=metal_arm_rec --valid_targets=metal,arm --optimize_out_type=naive_buffer
-
模型使用时(python代码调用)报错: [F 4/ 2 21:36:44.971 ...al/image_op/elementwise_image_compute.mm:218 setup_without_mps] elementwise_mul does not support the current input dimensions.
-
其他信息:
- 采用同样的转换和使用方法(METAL模式),对于mobilenet_v1模型(不包含elementwise_mul算子),不报错
- 模型转换和使用时不适用METAL模式,指定--valid_targets=arm,不报错
模型使用python代码片段:
2. Create paddle predictor
predictor = create_paddle_predictor(config)
3. Set input data
input_tensor = predictor.get_input(0)
c, h, w = 3, 48, 320 input_tensor.from_numpy(np.ones((1, c, h, w)).astype("float32")) predictor.run()
看代码报错,可能不支持当前的输入规模,能否手动调整一下呢
要怎么调整呢? 另外arm模式能正常运行,问题在metal代码上