MegEngine / MegCC

MegCC是一个运行时超轻量,高效,移植简单的深度学习模型编译器
Apache License 2.0
467 stars 55 forks source link

x86上mobilenetv3模型变慢 #33

Closed nicheng0019 closed 1 year ago

nicheng0019 commented 1 year ago

mobilenetv3 large模型,直接从onnx格式转成mge格式,使用megengine运行,在输入尺寸为320x320的情况下,大概需要36ms, 使用megcc的

mgb-to-tinynn mobilenetv3.mge ./x86 --enable_nchw44 --mgb_fuse_kernel

./runtime/scripts/runtime_build.py --kernel_dir ./x86

命令转成.tiny格式,用tinynn_test_lite测试耗时大概为86ms,请问这个可能是什么原因?多谢!

chenqy4933 commented 1 year ago

megcc 主要优化对象还是 Arm CPU,x86没有太关注性能,你在arm上测试一下呢?

nicheng0019 commented 1 year ago

好的,谢谢,我在arm上试一下