Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- licenses(["restricted"]) # NVIDIA proprietary license
- load("@local_config_cuda//cuda:build_defs.bzl", "if_version_equal_or_greater_than")
- exports_files([
- "version.txt",
- ])
- '''
- cc_import(
- name = "cupti_shared_library",
- hdrs = [":headers"],
- shared_library = "lib/libcupti.so.%{libcupti_version}",
- )
- '''
- cc_library(
- name = "cupti",
- #deps = [":cupti_shared_library"],
- visibility = ["//visibility:public"],
- )
- cc_library(
- name = "headers",
- #hdrs = glob([
- #"include/Openacc/**",
- #"include/Openmp/**",
- #"include/cuda_stdint.h",
- #"include/cupti.h",
- #"include/cupti_activity.h",
- #"include/cupti_activity_deprecated.h",
- #"include/cupti_callbacks.h",
- #"include/cupti_checkpoint.h",
- #"include/cupti_driver_cbid.h",
- #"include/cupti_events.h",
- #"include/cupti_metrics.h",
- #"include/cupti_nvtx_cbid.h",
- #"include/cupti_pcsampling.h",
- #"include/cupti_pcsampling_util.h",
- #"include/cupti_profiler_target.h",
- #"include/cupti_result.h",
- #"include/cupti_runtime_cbid.h",
- #"include/cupti_sass_metrics.h",
- #"include/cupti_target.h",
- #"include/cupti_version.h",
- #"include/generated_cudaGL_meta.h",
- #"include/generated_cudaVDPAU_meta.h",
- #"include/generated_cuda_gl_interop_meta.h",
- #"include/generated_cuda_meta.h",
- #"include/generated_cuda_runtime_api_meta.h",
- #"include/generated_cuda_vdpau_interop_meta.h",
- #"include/generated_cudart_removed_meta.h",
- #"include/generated_nvtx_meta.h",
- #"include/nvperf_common.h",
- #"include/nvperf_cuda_host.h",
- #"include/nvperf_host.h",
- #"include/nvperf_target.h",
- #]) + if_version_equal_or_greater_than(
- #"%{libcupti_minor_version}",
- #"2024.0",
- #["include/cupti_common.h"],
- #),
- include_prefix = "third_party/gpus/cuda/extras/CUPTI/include",
- includes = ["include/"],
- strip_include_prefix = "include",
- visibility = ["@local_config_cuda//cuda:__pkg__"],
- )
Advertisement
Add Comment
Please, Sign In to add comment