forked from tensorflow/tensorflow
-
Notifications
You must be signed in to change notification settings - Fork 96
/
Copy pathcudnn_frontend_header_fix.patch
52 lines (40 loc) · 1.58 KB
/
cudnn_frontend_header_fix.patch
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
diff --git a/include/cudnn_frontend.h b/include/cudnn_frontend.h
index e3f1ec8..373e52d 100644
--- a/include/cudnn_frontend.h
+++ b/include/cudnn_frontend.h
@@ -97,7 +97,7 @@
* - Simpler samples on how to use the new API.
*/
-#include <cudnn.h>
+#include "third_party/gpus/cudnn/cudnn.h"
#include "cudnn_frontend_ConvDesc.h"
#include "cudnn_frontend_Heuristics.h"
diff --git a/include/cudnn_frontend/backend/backend_descriptor.h b/include/cudnn_frontend/backend/backend_descriptor.h
index 47387a1..3864c1d 100644
--- a/include/cudnn_frontend/backend/backend_descriptor.h
+++ b/include/cudnn_frontend/backend/backend_descriptor.h
@@ -3,7 +3,7 @@
#include <memory>
#include "../graph_helpers.h"
-#include "cudnn.h"
+#include "third_party/gpus/cudnn/cudnn.h"
namespace cudnn_frontend::detail {
diff --git a/include/cudnn_frontend/backend/execution_helpers.h b/include/cudnn_frontend/backend/execution_helpers.h
index 334ffde..d2ca694 100644
--- a/include/cudnn_frontend/backend/execution_helpers.h
+++ b/include/cudnn_frontend/backend/execution_helpers.h
@@ -2,7 +2,7 @@
#include <vector>
-#include "cudnn.h"
+#include "third_party/gpus/cudnn/cudnn.h"
#include "backend_descriptor.h"
diff --git a/include/cudnn_frontend/backend/plan_helpers.h b/include/cudnn_frontend/backend/plan_helpers.h
index 1fa458d..8c37d10 100644
--- a/include/cudnn_frontend/backend/plan_helpers.h
+++ b/include/cudnn_frontend/backend/plan_helpers.h
@@ -2,7 +2,7 @@
#include <vector>
-#include "cudnn.h"
+#include "third_party/gpus/cudnn/cudnn.h"
#include "backend_descriptor.h"
#include "../knobs.h"