Paddle

Commit Graph

Author	SHA1	Message	Date
Chen Weihang	0b54d54fd8	Fix index overflow bug of the CUDA kernel loop increment (#25435 ) * fix softmax_with_cross_entropy cuda kernel overflow bug, test=develop * replace old macro & for condition, test=develop * polish details, test=develop	5 years ago
hutuxian	a6ce2306f9	support cvm_op run in gpu (#21300 ) Previously, CVM OP was only able to run in CPU. This PR implements its GPU kernel. What's more, we improve the UTs about CVM OP.	5 years ago

Author

SHA1

Message

Date

Chen Weihang

0b54d54fd8

Fix index overflow bug of the CUDA kernel loop increment (#25435 )

* fix softmax_with_cross_entropy cuda kernel overflow bug, test=develop

* replace old macro & for condition, test=develop

* polish details, test=develop

hutuxian

a6ce2306f9

support cvm_op run in gpu (#21300 )

Previously, CVM OP was only able to run in CPU. This PR implements its GPU kernel.
What's more, we improve the UTs about CVM OP.

2 Commits (824a79d383531e804e7274ef2141c30c7532e2c2)