Commit Graph

7 Commits (310edc0d0c9050f1f01c108655493c1935c00214)

Author SHA1 Message Date
danleifeng 9171f73714 fix fp16 grid_size for size=1; test=develop (#20812)
5 years ago
danleifeng 425279a57b Improve elementwise operators performance in same dimensions. (#19763)
5 years ago
Leo Chen 982e61f5ff Update elementwise double grad to save gpu memory (#19509)
6 years ago
Kaipeng Deng bd9bef5a4e
add elementwise_add_grad_grad op (#17366)
6 years ago
Yiqun Liu dcda20233c
Optimize the elementwise op using eigen (#15494)
6 years ago
Yu Yang 7604b1ad51 Fix Eigen macro when using GPU
6 years ago
Wu Yi a2d9b34417
Refine operator cmake (#14413)
6 years ago