CINN icon indicating copy to clipboard operation
CINN copied to clipboard

Published 20 hours ago • verified publisher icon PaddlePaddle

tests(auto-unroll): Improve the detection of Auto-Unroll

Open 6clc opened this issue 2 years ago • 0 comments

用新的测试框架增强auto-unroll测例。

该PR添加了2种cases: 1 ApplyOnMatmulWithUnroll：unroll矩阵乘的reduce轴，并检查block中的auto_unroll_max_step属性。 2 PureSpatial: 如果计算图中的算子类型均是Elementwise或者injective的，则不能进入auto-unroll

6clc avatar Mar 27 '23 08:03 6clc