Created by: zhangting2020
Performance optimization PR types
OPs PR changes
use eval to improve performance Describe
CPU Performance
op | input shape | p | before | after | speed up |
---|---|---|---|---|---|
dist_grad | [1, 64, 128, 128] | 2 | 1037.24 ms | 493.03 ms | 2倍 |
dist_grad | [1, 64, 128, 128] | -inf | 1002.08 ms | 490.838 ms | 2倍 |