Skip to content

Commit ccabdfe

Browse files
bstrinerezyang
authored andcommitted
Export getCudnnHandle (#7726)
1 parent 9243b64 commit ccabdfe

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

aten/src/ATen/cudnn/Handles.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
#pragma once
22

33
#include "cudnn-wrapper.h"
4+
#include "ATen/cuda/ATenCUDAGeneral.h"
45

56
namespace at { namespace native {
67

7-
cudnnHandle_t getCudnnHandle();
8+
AT_CUDA_API cudnnHandle_t getCudnnHandle();
89

910
}} // namespace

0 commit comments

Comments
 (0)