230 lines
7.0 KiB
Diff
230 lines
7.0 KiB
Diff
From 681ce638cbf9a115054071d7549f10cb0d3e37ac Mon Sep 17 00:00:00 2001
|
|
From: Kaixi Hou <kaixih@nvidia.com>
|
|
Date: Tue, 4 May 2021 15:21:11 -0700
|
|
Subject: [PATCH] Update headers path to TF-compat
|
|
|
|
---
|
|
include/cudnn_backend_base.h | 2 +-
|
|
include/cudnn_frontend_ConvDesc.h | 4 ++--
|
|
include/cudnn_frontend_Engine.h | 4 ++--
|
|
include/cudnn_frontend_EngineConfig.h | 4 ++--
|
|
include/cudnn_frontend_EngineFallbackList.h | 2 +-
|
|
include/cudnn_frontend_ExecutionPlan.h | 4 ++--
|
|
include/cudnn_frontend_Filters.h | 2 +-
|
|
include/cudnn_frontend_Heuristics.h | 4 ++--
|
|
include/cudnn_frontend_MatMulDesc.h | 4 ++--
|
|
include/cudnn_frontend_Operation.h | 4 ++--
|
|
include/cudnn_frontend_OperationGraph.h | 4 ++--
|
|
include/cudnn_frontend_PointWiseDesc.h | 4 ++--
|
|
include/cudnn_frontend_ReductionDesc.h | 4 ++--
|
|
include/cudnn_frontend_VariantPack.h | 4 ++--
|
|
14 files changed, 25 insertions(+), 25 deletions(-)
|
|
|
|
diff --git a/include/cudnn_backend_base.h b/include/cudnn_backend_base.h
|
|
index 56d8bec..8ceb19c 100644
|
|
--- a/include/cudnn_backend_base.h
|
|
+++ b/include/cudnn_backend_base.h
|
|
@@ -24,7 +24,7 @@
|
|
|
|
#include <ostream>
|
|
|
|
-#include <cudnn.h>
|
|
+#include "third_party/gpus/cudnn/cudnn.h"
|
|
|
|
namespace cudnn_frontend {
|
|
|
|
diff --git a/include/cudnn_frontend_ConvDesc.h b/include/cudnn_frontend_ConvDesc.h
|
|
index 871822b..115d542 100644
|
|
--- a/include/cudnn_frontend_ConvDesc.h
|
|
+++ b/include/cudnn_frontend_ConvDesc.h
|
|
@@ -29,8 +29,8 @@
|
|
#include <sstream>
|
|
#include <utility>
|
|
|
|
-#include <cudnn.h>
|
|
-#include <cudnn_backend.h>
|
|
+#include "third_party/gpus/cudnn/cudnn.h"
|
|
+#include "third_party/gpus/cudnn/cudnn_backend.h"
|
|
|
|
#include "cudnn_frontend_utils.h"
|
|
|
|
diff --git a/include/cudnn_frontend_Engine.h b/include/cudnn_frontend_Engine.h
|
|
index 83d3f80..a1b0301 100644
|
|
--- a/include/cudnn_frontend_Engine.h
|
|
+++ b/include/cudnn_frontend_Engine.h
|
|
@@ -30,8 +30,8 @@
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
-#include <cudnn.h>
|
|
-#include <cudnn_backend.h>
|
|
+#include "third_party/gpus/cudnn/cudnn.h"
|
|
+#include "third_party/gpus/cudnn/cudnn_backend.h"
|
|
|
|
#include "cudnn_frontend_OperationGraph.h"
|
|
#include "cudnn_frontend_utils.h"
|
|
diff --git a/include/cudnn_frontend_EngineConfig.h b/include/cudnn_frontend_EngineConfig.h
|
|
index a9cdb7f..0bb47ae 100644
|
|
--- a/include/cudnn_frontend_EngineConfig.h
|
|
+++ b/include/cudnn_frontend_EngineConfig.h
|
|
@@ -29,8 +29,8 @@
|
|
#include <sstream>
|
|
#include <utility>
|
|
|
|
-#include <cudnn.h>
|
|
-#include <cudnn_backend.h>
|
|
+#include "third_party/gpus/cudnn/cudnn.h"
|
|
+#include "third_party/gpus/cudnn/cudnn_backend.h"
|
|
|
|
#include "cudnn_frontend_Engine.h"
|
|
#include "cudnn_frontend_utils.h"
|
|
diff --git a/include/cudnn_frontend_EngineFallbackList.h b/include/cudnn_frontend_EngineFallbackList.h
|
|
index 323106a..d90a1ea 100644
|
|
--- a/include/cudnn_frontend_EngineFallbackList.h
|
|
+++ b/include/cudnn_frontend_EngineFallbackList.h
|
|
@@ -22,7 +22,7 @@
|
|
|
|
#pragma once
|
|
|
|
-#include <cudnn.h>
|
|
+#include "third_party/gpus/cudnn/cudnn.h"
|
|
#include <numeric>
|
|
#include "cudnn_frontend_Heuristics.h"
|
|
|
|
diff --git a/include/cudnn_frontend_ExecutionPlan.h b/include/cudnn_frontend_ExecutionPlan.h
|
|
index 42fd09b..f4a294f 100644
|
|
--- a/include/cudnn_frontend_ExecutionPlan.h
|
|
+++ b/include/cudnn_frontend_ExecutionPlan.h
|
|
@@ -29,8 +29,8 @@
|
|
#include <sstream>
|
|
#include <utility>
|
|
|
|
-#include <cudnn.h>
|
|
-#include <cudnn_backend.h>
|
|
+#include "third_party/gpus/cudnn/cudnn.h"
|
|
+#include "third_party/gpus/cudnn/cudnn_backend.h"
|
|
|
|
#include "cudnn_frontend_Engine.h"
|
|
#include "cudnn_frontend_utils.h"
|
|
diff --git a/include/cudnn_frontend_Filters.h b/include/cudnn_frontend_Filters.h
|
|
index b244766..3e9273b 100644
|
|
--- a/include/cudnn_frontend_Filters.h
|
|
+++ b/include/cudnn_frontend_Filters.h
|
|
@@ -22,7 +22,7 @@
|
|
|
|
#pragma once
|
|
|
|
-#include <cudnn.h>
|
|
+#include "third_party/gpus/cudnn/cudnn.h"
|
|
|
|
namespace cudnn_frontend {
|
|
|
|
diff --git a/include/cudnn_frontend_Heuristics.h b/include/cudnn_frontend_Heuristics.h
|
|
index 3d0dacd..fec5e31 100644
|
|
--- a/include/cudnn_frontend_Heuristics.h
|
|
+++ b/include/cudnn_frontend_Heuristics.h
|
|
@@ -25,8 +25,8 @@
|
|
#include <vector>
|
|
#include <mutex>
|
|
|
|
-#include <cudnn.h>
|
|
-#include <cudnn_backend.h>
|
|
+#include "third_party/gpus/cudnn/cudnn.h"
|
|
+#include "third_party/gpus/cudnn/cudnn_backend.h"
|
|
|
|
#include "cudnn_frontend_OperationGraph.h"
|
|
#include "cudnn_frontend_EngineConfig.h"
|
|
diff --git a/include/cudnn_frontend_MatMulDesc.h b/include/cudnn_frontend_MatMulDesc.h
|
|
index 5f3161a..c357cc1 100644
|
|
--- a/include/cudnn_frontend_MatMulDesc.h
|
|
+++ b/include/cudnn_frontend_MatMulDesc.h
|
|
@@ -29,8 +29,8 @@
|
|
#include <sstream>
|
|
#include <utility>
|
|
|
|
-#include <cudnn.h>
|
|
-#include <cudnn_backend.h>
|
|
+#include "third_party/gpus/cudnn/cudnn.h"
|
|
+#include "third_party/gpus/cudnn/cudnn_backend.h"
|
|
|
|
#include "cudnn_frontend_utils.h"
|
|
|
|
diff --git a/include/cudnn_frontend_Operation.h b/include/cudnn_frontend_Operation.h
|
|
index 4d4e2dc..32fd1b1 100644
|
|
--- a/include/cudnn_frontend_Operation.h
|
|
+++ b/include/cudnn_frontend_Operation.h
|
|
@@ -29,8 +29,8 @@
|
|
#include <sstream>
|
|
#include <utility>
|
|
|
|
-#include <cudnn.h>
|
|
-#include <cudnn_backend.h>
|
|
+#include "third_party/gpus/cudnn/cudnn.h"
|
|
+#include "third_party/gpus/cudnn/cudnn_backend.h"
|
|
|
|
#include "cudnn_frontend_ConvDesc.h"
|
|
#include "cudnn_frontend_PointWiseDesc.h"
|
|
diff --git a/include/cudnn_frontend_OperationGraph.h b/include/cudnn_frontend_OperationGraph.h
|
|
index b240496..756f67e 100644
|
|
--- a/include/cudnn_frontend_OperationGraph.h
|
|
+++ b/include/cudnn_frontend_OperationGraph.h
|
|
@@ -30,8 +30,8 @@
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
-#include <cudnn.h>
|
|
-#include <cudnn_backend.h>
|
|
+#include "third_party/gpus/cudnn/cudnn.h"
|
|
+#include "third_party/gpus/cudnn/cudnn_backend.h"
|
|
|
|
#include "cudnn_frontend_Operation.h"
|
|
#include "cudnn_frontend_utils.h"
|
|
diff --git a/include/cudnn_frontend_PointWiseDesc.h b/include/cudnn_frontend_PointWiseDesc.h
|
|
index bf0ad54..869bbfd 100644
|
|
--- a/include/cudnn_frontend_PointWiseDesc.h
|
|
+++ b/include/cudnn_frontend_PointWiseDesc.h
|
|
@@ -30,8 +30,8 @@
|
|
#include <utility>
|
|
#include <limits>
|
|
|
|
-#include <cudnn.h>
|
|
-#include <cudnn_backend.h>
|
|
+#include "third_party/gpus/cudnn/cudnn.h"
|
|
+#include "third_party/gpus/cudnn/cudnn_backend.h"
|
|
|
|
#include "cudnn_frontend_utils.h"
|
|
|
|
diff --git a/include/cudnn_frontend_ReductionDesc.h b/include/cudnn_frontend_ReductionDesc.h
|
|
index 046af46..f82a643 100644
|
|
--- a/include/cudnn_frontend_ReductionDesc.h
|
|
+++ b/include/cudnn_frontend_ReductionDesc.h
|
|
@@ -29,8 +29,8 @@
|
|
#include <sstream>
|
|
#include <utility>
|
|
|
|
-#include <cudnn.h>
|
|
-#include <cudnn_backend.h>
|
|
+#include "third_party/gpus/cudnn/cudnn.h"
|
|
+#include "third_party/gpus/cudnn/cudnn_backend.h"
|
|
|
|
#include "cudnn_frontend_utils.h"
|
|
|
|
diff --git a/include/cudnn_frontend_VariantPack.h b/include/cudnn_frontend_VariantPack.h
|
|
index 0fe4f2f..41a2552 100644
|
|
--- a/include/cudnn_frontend_VariantPack.h
|
|
+++ b/include/cudnn_frontend_VariantPack.h
|
|
@@ -30,8 +30,8 @@
|
|
#include <sstream>
|
|
#include <utility>
|
|
|
|
-#include <cudnn.h>
|
|
-#include <cudnn_backend.h>
|
|
+#include "third_party/gpus/cudnn/cudnn.h"
|
|
+#include "third_party/gpus/cudnn/cudnn_backend.h"
|
|
|
|
#include "cudnn_frontend_utils.h"
|
|
|
|
--
|
|
2.25.1
|
|
|