frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BE] [cuDNN] Always build assuming cuDNN >= 8.0 (#95722)

Browse Source 
<!--
copilot:summary
-->
### <samp>🤖 Generated by Copilot at 27084ed</samp>

This pull request simplifies and cleans up the code that uses the cuDNN library for convolution, batch normalization, CTC loss, and quantized operations. It removes the unnecessary checks and conditions for older cuDNN versions and the experimental cuDNN v8 API, and ~~replaces them with the stable `cudnn_frontend` API that requires cuDNN v8 or higher. It also adds the dependency and configuration for the `cudnn_frontend` library in the cmake and bazel files.~~ Correction: The v7 API will still be available with this PR, and can still be used, without any changes to the defaults. This change simply always _builds_ the v8 API, and removes the case where _only_ the v7 API is built.

This is a re-land of https://github.com/pytorch/pytorch/pull/91527

Pull Request resolved: https://github.com/pytorch/pytorch/pull/95722
Approved by: https://github.com/malfet
This commit is contained in:
Eddie Yan
2023-11-08 07:53:23 +00:00
committed by PyTorch MergeBot
parent 8ba11bf79d
commit df4f0b3829
24 changed files with 55 additions and 170 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
WORKSPACE
View File

@ -246,6 +246,12 @@ new_local_repository(
path = "/usr/",
)
new_local_repository(
name = "cudnn_frontend",
build_file = "@//third_party:cudnn_frontend.BUILD",
path = "third_party/cudnn_frontend/",
)
local_repository(
name = "com_github_google_flatbuffers",
path = "third_party/flatbuffers",