mirror of
https://github.com/zebrajr/tensorflow.git
synced 2026-01-15 12:15:41 +00:00
53f9a17c0e
Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/46965 This PR integrates the new cuDNN Frontend API (https://github.com/NVIDIA/cudnn-frontend) for the convolution backends. FYI. @nluehr Copybara import of the project: --61ec89a3e8by Kaixi Hou <kaixih@nvidia.com>: CUDNNv8: Add configs for headers --d3c34717feby Kaixi Hou <kaixih@nvidia.com>: CUDNNv8: Add support in stream executor --5d54a49eccby Kaixi Hou <kaixih@nvidia.com>: CUDNNv8: Add support in autotuner --1ee715d6eeby Kaixi Hou <kaixih@nvidia.com>: CUDNNv8: Add support in Conv2D --2badb9e39eby Kaixi Hou <kaixih@nvidia.com>: CUDNNv8: Add support in Conv3D --76b4645547by Kaixi Hou <kaixih@nvidia.com>: CUDNNv8: Fix a failed test --322e858bd8by Kaixi Hou <kaixih@nvidia.com>: Cleanup codes --075baf8fd6by Kaixi Hou <kaixih@nvidia.com>: Merge AutotuneExecutionPlanResult into AutotuneResult --fd2335ed63by Kaixi Hou <kaixih@nvidia.com>: Use AlgorithmXXX instead of ExecutionPlanXXX --aa6f984dc7by Kaixi Hou <kaixih@nvidia.com>: Remove ProfileExecutionPlanResults --994af73b61by Kaixi Hou <kaixih@nvidia.com>: Separate the cudnn-specific DoConvolve --ed22ef7896by Kaixi Hou <kaixih@nvidia.com>: Cleanup code --a047609b55by Kaixi Hou <kaixih@nvidia.com>: Add env var to switch on cudnn frontend --d5470a4018by Kaixi Hou <kaixih@nvidia.com>: Cleanup code --558472bcdaby Kaixi Hou <kaixih@nvidia.com>: Unify autotuning --4e780fed91by Kaixi Hou <kaixih@nvidia.com>: Remove some macros --c865c2d882by Kaixi Hou <kaixih@nvidia.com>: Combine BestCudnnConvAlgorithm --a51a1ea7f1by Kaixi Hou <kaixih@nvidia.com>: Use original autotuning map --a114edefdbby Kaixi Hou <kaixih@nvidia.com>: Use shared ptr --7e8898c42fby Kaixi Hou <kaixih@nvidia.com>: Remove some macros --5ffdb1f896by Kaixi Hou <kaixih@nvidia.com>: Use if macros --6838420da8by Kaixi Hou <kaixih@nvidia.com>: Remove if macros PiperOrigin-RevId: 364409851 Change-Id: I829affdf6a2fb344a16e82ab201ff76b7421eef1
25 lines
492 B
Plaintext
25 lines
492 B
Plaintext
# Description:
|
|
# The cuDNN Frontend API is a C++ header-only library that demonstrates how
|
|
# to use the cuDNN C backend API.
|
|
|
|
package(
|
|
default_visibility = ["//visibility:public"],
|
|
)
|
|
|
|
licenses(["notice"]) # MIT
|
|
|
|
exports_files(["LICENSE.txt"])
|
|
|
|
filegroup(
|
|
name = "cudnn_frontend_header_files",
|
|
srcs = glob([
|
|
"include/**",
|
|
]),
|
|
)
|
|
|
|
cc_library(
|
|
name = "cudnn_frontend",
|
|
hdrs = [":cudnn_frontend_header_files"],
|
|
include_prefix = "third_party/cudnn_frontend",
|
|
)
|