直接计算conv2d的cuda算子实现
详细介绍参考：https://zhuanlan.zhihu.com/p/613538649
需要确保本机有cuda和cudnn
nvcc main.cu -o test -lcudnn
如果找不到cudnn请手动指定目录
forked from Pegessi/conv2d_direct
-
Notifications
You must be signed in to change notification settings - Fork 0
ckl117/conv2d_direct
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Cuda 100.0%