Caffe2 - C++ API
A deep learning, cross platform ML framework
cuda_cudnn_test.cpp
1 #include <gtest/gtest.h>
2 
3 #include <ATen/ATen.h>
4 #include <ATen/cuda/CUDAContext.h>
5 #include <ATen/cudnn/Descriptors.h>
6 #include <ATen/cudnn/Handle.h>
7 
8 using namespace at;
9 using namespace at::native;
10 
11 TEST(CUDNNTest, CUDNNTestCUDA) {
12  if (!at::cuda::is_available()) return;
13  manual_seed(123);
14 }
Flush-To-Zero and Denormals-Are-Zero mode.