2024-02-14 18:53:01 +00:00
|
|
|
diff --git a/include/cudnn_frontend.h b/include/cudnn_frontend.h
|
2024-11-04 17:00:42 +00:00
|
|
|
index e3f1ec8..373e52d 100644
|
2024-02-14 18:53:01 +00:00
|
|
|
--- a/include/cudnn_frontend.h
|
|
|
|
+++ b/include/cudnn_frontend.h
|
|
|
|
@@ -97,7 +97,7 @@
|
|
|
|
* - Simpler samples on how to use the new API.
|
|
|
|
*/
|
2021-03-22 21:32:07 +00:00
|
|
|
|
|
|
|
-#include <cudnn.h>
|
|
|
|
+#include "third_party/gpus/cudnn/cudnn.h"
|
|
|
|
|
|
|
|
#include "cudnn_frontend_ConvDesc.h"
|
2024-02-14 18:53:01 +00:00
|
|
|
#include "cudnn_frontend_Heuristics.h"
|
2024-06-21 14:10:45 +00:00
|
|
|
diff --git a/include/cudnn_frontend/backend/backend_descriptor.h b/include/cudnn_frontend/backend/backend_descriptor.h
|
2024-11-04 17:00:42 +00:00
|
|
|
index 47387a1..3864c1d 100644
|
2024-06-21 14:10:45 +00:00
|
|
|
--- a/include/cudnn_frontend/backend/backend_descriptor.h
|
|
|
|
+++ b/include/cudnn_frontend/backend/backend_descriptor.h
|
2024-11-04 17:00:42 +00:00
|
|
|
@@ -3,7 +3,7 @@
|
2024-06-21 14:10:45 +00:00
|
|
|
#include <memory>
|
|
|
|
|
2024-11-04 17:00:42 +00:00
|
|
|
#include "../graph_helpers.h"
|
2024-06-21 14:10:45 +00:00
|
|
|
-#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"
|
|
|
|
|