Skip to content

cudnn samples are provided to verify if the operator's calculation results are correct on other gpu products(e.g., NPU, TPU) compatible to NVIDIA cuda/cudnn.

Notifications You must be signed in to change notification settings

HangJie720/cudnn-samples

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

cudnn-samples

cudnn samples are provided to verify if the operator's calculation results are correct on other gpu products compatible to nvidia cuda/cudnn.

how to use

for example, we try to generate data(e.g, input/weight/bias/outout) with tensorflow api, such as tf.nn.maxpool(), and then using this data to verify the correctness of cudnnMaxpoolForward api.

# generate input/output data 
$ python tf-maxpooling.py
# modify float32.h to replace input/output data.
dtype input[IN_SIZE] ={...}
dtype output[OUT_SIZE] = {...}
# verify cudnnMaxpoolForward api.
$ make
$ ./test

About

cudnn samples are provided to verify if the operator's calculation results are correct on other gpu products(e.g., NPU, TPU) compatible to NVIDIA cuda/cudnn.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published