From 4b036aa5bc53cefb69983d2fab759b4d5a43e0ba Mon Sep 17 00:00:00 2001 From: Kaixi Hou Date: Mon, 4 Nov 2024 09:00:42 -0800 Subject: [PATCH] PR #18839: [NVIDIA] Update Cudnn Frontend to 1.8.0 Imported from GitHub PR https://github.com/openxla/xla/pull/18839 This PR bumps up the cudnn frontend version to 1.8.0. Copybara import of the project: -- 4314cffd7f3a7aac234aa7d484173c56ed360af3 by kaixih : Update cudnn frontend to 1.8.0 Merging this change closes #18839 PiperOrigin-RevId: 692985487 --- tensorflow/workspace2.bzl | 6 +++--- third_party/cudnn_frontend_header_fix.patch | 8 ++++---- .../xla/third_party/cudnn_frontend_header_fix.patch | 8 ++++---- third_party/xla/workspace2.bzl | 6 +++--- 4 files changed, 14 insertions(+), 14 deletions(-) diff --git a/tensorflow/workspace2.bzl b/tensorflow/workspace2.bzl index fedad18909d..598dfd5cdf8 100644 --- a/tensorflow/workspace2.bzl +++ b/tensorflow/workspace2.bzl @@ -190,9 +190,9 @@ def _tf_repositories(): name = "cudnn_frontend_archive", build_file = "//third_party:cudnn_frontend.BUILD", patch_file = ["//third_party:cudnn_frontend_header_fix.patch"], - sha256 = "bd1037f8e7218d0d44ff2ff11d0c95175a5a27a82d8ea92879e23eafd6d5df02", - strip_prefix = "cudnn-frontend-1.6.1", - urls = tf_mirror_urls("https://github.com/NVIDIA/cudnn-frontend/archive/refs/tags/v1.6.1.zip"), + sha256 = "5f77784dc3ccbca7aca5ea0b5a6e31b95aa85023c5942d22be5fa8dd6c339d81", + strip_prefix = "cudnn-frontend-1.8.0", + urls = tf_mirror_urls("https://github.com/NVIDIA/cudnn-frontend/archive/refs/tags/v1.8.0.zip"), ) tf_http_archive( diff --git a/third_party/cudnn_frontend_header_fix.patch b/third_party/cudnn_frontend_header_fix.patch index 6a57589b53e..898be6d6bd4 100644 --- a/third_party/cudnn_frontend_header_fix.patch +++ b/third_party/cudnn_frontend_header_fix.patch @@ -1,5 +1,5 @@ diff --git a/include/cudnn_frontend.h b/include/cudnn_frontend.h -index 7bca425..cd74e3a 100644 +index e3f1ec8..373e52d 100644 --- a/include/cudnn_frontend.h +++ b/include/cudnn_frontend.h @@ -97,7 +97,7 @@ @@ -12,13 +12,13 @@ index 7bca425..cd74e3a 100644 #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 dc7ad25..ade164a 100644 +index 47387a1..3864c1d 100644 --- a/include/cudnn_frontend/backend/backend_descriptor.h +++ b/include/cudnn_frontend/backend/backend_descriptor.h -@@ -2,7 +2,7 @@ - +@@ -3,7 +3,7 @@ #include + #include "../graph_helpers.h" -#include "cudnn.h" +#include "third_party/gpus/cudnn/cudnn.h" diff --git a/third_party/xla/third_party/cudnn_frontend_header_fix.patch b/third_party/xla/third_party/cudnn_frontend_header_fix.patch index 6a57589b53e..898be6d6bd4 100644 --- a/third_party/xla/third_party/cudnn_frontend_header_fix.patch +++ b/third_party/xla/third_party/cudnn_frontend_header_fix.patch @@ -1,5 +1,5 @@ diff --git a/include/cudnn_frontend.h b/include/cudnn_frontend.h -index 7bca425..cd74e3a 100644 +index e3f1ec8..373e52d 100644 --- a/include/cudnn_frontend.h +++ b/include/cudnn_frontend.h @@ -97,7 +97,7 @@ @@ -12,13 +12,13 @@ index 7bca425..cd74e3a 100644 #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 dc7ad25..ade164a 100644 +index 47387a1..3864c1d 100644 --- a/include/cudnn_frontend/backend/backend_descriptor.h +++ b/include/cudnn_frontend/backend/backend_descriptor.h -@@ -2,7 +2,7 @@ - +@@ -3,7 +3,7 @@ #include + #include "../graph_helpers.h" -#include "cudnn.h" +#include "third_party/gpus/cudnn/cudnn.h" diff --git a/third_party/xla/workspace2.bzl b/third_party/xla/workspace2.bzl index 64ade49a93d..f8ed373e790 100644 --- a/third_party/xla/workspace2.bzl +++ b/third_party/xla/workspace2.bzl @@ -52,9 +52,9 @@ def _tf_repositories(): name = "cudnn_frontend_archive", build_file = "//third_party:cudnn_frontend.BUILD", patch_file = ["//third_party:cudnn_frontend_header_fix.patch"], - sha256 = "bd1037f8e7218d0d44ff2ff11d0c95175a5a27a82d8ea92879e23eafd6d5df02", - strip_prefix = "cudnn-frontend-1.6.1", - urls = tf_mirror_urls("https://github.com/NVIDIA/cudnn-frontend/archive/refs/tags/v1.6.1.zip"), + sha256 = "5f77784dc3ccbca7aca5ea0b5a6e31b95aa85023c5942d22be5fa8dd6c339d81", + strip_prefix = "cudnn-frontend-1.8.0", + urls = tf_mirror_urls("https://github.com/NVIDIA/cudnn-frontend/archive/refs/tags/v1.8.0.zip"), ) tf_http_archive(