xref: /aosp_15_r20/external/tensorflow/third_party/cudnn_frontend.BUILD (revision b6fb3261f9314811a0f4371741dbb8839866f948)
1*b6fb3261SAndroid Build Coastguard Worker# Description:
2*b6fb3261SAndroid Build Coastguard Worker# The cuDNN Frontend API is a C++ header-only library that demonstrates how
3*b6fb3261SAndroid Build Coastguard Worker# to use the cuDNN C backend API.
4*b6fb3261SAndroid Build Coastguard Worker
5*b6fb3261SAndroid Build Coastguard Workerpackage(
6*b6fb3261SAndroid Build Coastguard Worker    default_visibility = ["//visibility:public"],
7*b6fb3261SAndroid Build Coastguard Worker)
8*b6fb3261SAndroid Build Coastguard Worker
9*b6fb3261SAndroid Build Coastguard Workerlicenses(["notice"])  # MIT
10*b6fb3261SAndroid Build Coastguard Worker
11*b6fb3261SAndroid Build Coastguard Workerexports_files(["LICENSE.txt"])
12*b6fb3261SAndroid Build Coastguard Worker
13*b6fb3261SAndroid Build Coastguard Workerfilegroup(
14*b6fb3261SAndroid Build Coastguard Worker    name = "cudnn_frontend_header_files",
15*b6fb3261SAndroid Build Coastguard Worker    srcs = glob([
16*b6fb3261SAndroid Build Coastguard Worker        "include/**",
17*b6fb3261SAndroid Build Coastguard Worker    ]),
18*b6fb3261SAndroid Build Coastguard Worker)
19*b6fb3261SAndroid Build Coastguard Worker
20*b6fb3261SAndroid Build Coastguard Workercc_library(
21*b6fb3261SAndroid Build Coastguard Worker    name = "cudnn_frontend",
22*b6fb3261SAndroid Build Coastguard Worker    hdrs = [":cudnn_frontend_header_files"],
23*b6fb3261SAndroid Build Coastguard Worker    include_prefix = "third_party/cudnn_frontend",
24*b6fb3261SAndroid Build Coastguard Worker)
25