Skip to content

Commit 1ff436e

Browse files
committed
Fixing header for CPU compilation
1 parent 2772dca commit 1ff436e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

dlib/cuda/cudnn_dlibapi.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,10 @@
44
#define DLIB_DNN_CuDNN_H_
55

66
#ifdef DLIB_USE_CUDA
7-
87
#include "cuda_errors.h"
98
#include <memory>
109
#include "cuda_data_ptr.h"
10+
#endif // DLIB_USE_CUDA
1111

1212
namespace dlib
1313
{
@@ -25,6 +25,7 @@ namespace dlib
2525
!*/
2626
enum class operation_mode { CHANNEL_WISE = 0, PLANE_WISE = 1 };
2727

28+
#ifdef DLIB_USE_CUDA
2829
namespace cuda
2930
{
3031

@@ -429,9 +430,8 @@ namespace dlib
429430
// ------------------------------------------------------------------------------------
430431

431432
}
432-
}
433-
434433
#endif // DLIB_USE_CUDA
434+
}
435435

436436
#endif // DLIB_DNN_CuDNN_H_
437437

0 commit comments

Comments
 (0)