src/runtime/contrib/cudnn · e3eff20d8a91c2b986548f780493a8e30b0c2b7a · wenyuanbo / tic

[Perf] Enhance cudnn and cublas backend and enable TensorCore (#4353) · dabde40f

* add half and mix precision support to cublas backend

* add TensorCore support in CuDNN

* enhance CuDNN support

* address comments and fix lint

* fix

* add fp16 test

committed Nov 24, 2019

dabde40f

Name	Last commit	Last update
..
conv_forward.cc		Loading commit data...
cudnn_utils.cc		Loading commit data...
cudnn_utils.h		Loading commit data...