Commit Graph

9 Commits (291aa23153a27ac1fd21b506cec163e78bb428a7)

Author SHA1 Message Date
Kexin Zhao d00bd9eb72 Update the cuda API and enable tensor core for GEMM (#9622)
8 years ago
Kexin Zhao ed2bc194c5
Merge pull request #9176 from kexinzhao/batch_norm_fp16
8 years ago
Kexin Zhao 39c676e208 initial commit
8 years ago
yangyaming bf3f56e899 Finish adaption for backward.
8 years ago
Kexin Zhao 3b44b849d3 address comments
8 years ago
kexinzhao 90215b7844
Add float16 GEMM math function on GPU (#8695)
8 years ago
qingqing01 24509f4af9 Fix the grammar in copyright. (#8403)
8 years ago
Yi Wang fc374821dd Correct #include path
8 years ago
Yi Wang 90648f336d Move file to fluid/; Edit CMakeLists.txt
8 years ago