Commit Graph

7 Commits (98b59cb82ce4a84d70f65dd7fd9f2ebcafb5daa9)

Author SHA1 Message Date
danleifeng 9171f73714 fix fp16 grid_size for size=1; test=develop (#20812)
6 years ago
danleifeng 425279a57b Improve elementwise operators performance in same dimensions. (#19763)
6 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
7 years ago
Wu Yi a2d9b34417
Refine operator cmake (#14413)
7 years ago