pytorch

1
# Adopted from: https://github.com/tensorflow/tensorflow/blob/master/third_party/cudnn_frontend.BUILD
2

3
# Description:
4
# The cuDNN Frontend API is a C++ header-only library that demonstrates how
5
# to use the cuDNN C backend API.
6

7
load("@rules_cc//cc:defs.bzl", "cc_library")
8

9
package(
10
    default_visibility = ["//visibility:public"],
11
)
12

13
licenses(["notice"])  # MIT
14

15
exports_files(["LICENSE.txt"])
16

17
cc_library(
18
    name = "cudnn_frontend",
19
    hdrs = glob(["include/**"]),
20
    includes = ["include/"],
21
    include_prefix = "third_party/cudnn_frontend",
22
)
23