tensorflow/third_party/nvtx.BUILD
Frederic Bastien e353a2748f Add NVTX to XLA.
This reuse the same string as cupti, which is human readable, but was
more intended to be read by machine.

Should also works for TF but not tested.
2022-12-12 18:29:16 +00:00

21 lines
418 B
Plaintext

#Description : NVIDIA Tools Extension (NVTX) library for adding profiling annotations to applications.
package(
default_visibility = ["//visibility:public"],
)
licenses(["restricted"]) # NVIDIA proprietary license
filegroup(
name = "nvtx_header_files",
srcs = glob([
"nvtx3/**",
]),
)
cc_library(
name = "nvtx",
hdrs = [":nvtx_header_files"],
include_prefix = "third_party",
)