Skip to content

P
Paddle

PaddlePaddle / Paddle
1 年多 前同步成功

通知 2301
Star 20931
Fork 5422
P
Paddle

Speedup reader check_input_array when item is array !25395

!25395 已合并 由 saxon_zh@saxon_zh 创建

Created by: wangxicoding

PR types

Performance optimization

PR changes

Others

Describe

When the item is ndarry, _check_input_array will copy it as a new array, this copy is unnecessary. Now use asarray to optimize this scene.

image As shown in the figure, when the amount of data is 100 * 300 * 768 * 4 = 87.89MB, it takes 77.7ms to copy. While asarray uses zerocopy in this scenario, which takes only 326ns.