xref: /aosp_15_r20/external/pytorch/third_party/cudnn_frontend.BUILD (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker# Adopted from: https://github.com/tensorflow/tensorflow/blob/master/third_party/cudnn_frontend.BUILD
2*da0073e9SAndroid Build Coastguard Worker
3*da0073e9SAndroid Build Coastguard Worker# Description:
4*da0073e9SAndroid Build Coastguard Worker# The cuDNN Frontend API is a C++ header-only library that demonstrates how
5*da0073e9SAndroid Build Coastguard Worker# to use the cuDNN C backend API.
6*da0073e9SAndroid Build Coastguard Worker
7*da0073e9SAndroid Build Coastguard Workerload("@rules_cc//cc:defs.bzl", "cc_library")
8*da0073e9SAndroid Build Coastguard Worker
9*da0073e9SAndroid Build Coastguard Workerpackage(
10*da0073e9SAndroid Build Coastguard Worker    default_visibility = ["//visibility:public"],
11*da0073e9SAndroid Build Coastguard Worker)
12*da0073e9SAndroid Build Coastguard Worker
13*da0073e9SAndroid Build Coastguard Workerlicenses(["notice"])  # MIT
14*da0073e9SAndroid Build Coastguard Worker
15*da0073e9SAndroid Build Coastguard Workerexports_files(["LICENSE.txt"])
16*da0073e9SAndroid Build Coastguard Worker
17*da0073e9SAndroid Build Coastguard Workercc_library(
18*da0073e9SAndroid Build Coastguard Worker    name = "cudnn_frontend",
19*da0073e9SAndroid Build Coastguard Worker    hdrs = glob(["include/**"]),
20*da0073e9SAndroid Build Coastguard Worker    includes = ["include/"],
21*da0073e9SAndroid Build Coastguard Worker    include_prefix = "third_party/cudnn_frontend",
22*da0073e9SAndroid Build Coastguard Worker)
23