Paddle matmul mkl中infer shape bug

bttbmeg0  于 2021-11-30  发布在  Java
关注(0)|答案(0)|浏览(148)

使用paddle cpu-mkl版本 https://paddle-wheel.bj.bcebos.com/latest-cpu-mkl/paddlepaddle-latest-cp27-cp27mu-linux_x86_64.whl
最小复现脚本:

报错:

原因是在mkl环境下https://github.com/PaddlePaddle/Paddle/blob/develop/paddle/fluid/operators/matmul_op.cc#L314 matmul的infer shape有误,需要修改

暂无答案!

目前还没有任何答案,快来回答吧!

相关问题