Paddle-Lite icon indicating copy to clipboard operation
Paddle-Lite copied to clipboard

elementwise_mul算子无法在METAL环境正常使用

Open JKyang09 opened this issue 8 months ago • 2 comments

  • 版本、环境信息:    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.

  • 其他信息:

  1. 采用同样的转换和使用方法(METAL模式),对于mobilenet_v1模型(不包含elementwise_mul算子),不报错
  2. 模型转换和使用时不适用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()

JKyang09 avatar Apr 03 '25 03:04 JKyang09

看代码报错,可能不支持当前的输入规模,能否手动调整一下呢

iosmers avatar Apr 10 '25 01:04 iosmers

要怎么调整呢? 另外arm模式能正常运行,问题在metal代码上

JKyang09 avatar Apr 11 '25 02:04 JKyang09