Skip to content

Instantly share code, notes, and snippets.

View tohnee's full-sized avatar
🏠
Working from home

tohnee tohnee

🏠
Working from home
  • 安徽
View GitHub Profile
@goldsborough
goldsborough / conv.cu
Last active February 2, 2025 09:14
Convolution with cuDNN
#include <cudnn.h>
#include <cassert>
#include <cstdlib>
#include <iostream>
#include <opencv2/opencv.hpp>
#define checkCUDNN(expression) \
{ \
cudnnStatus_t status = (expression); \
if (status != CUDNN_STATUS_SUCCESS) { \