daskol

BUILD

Oct 22nd, 2024
229
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 2.12 KB | Source Code | 0 0
  1. licenses(["restricted"])  # NVIDIA proprietary license
  2. load("@local_config_cuda//cuda:build_defs.bzl", "if_version_equal_or_greater_than")
  3. exports_files([
  4.     "version.txt",
  5. ])
  6. '''
  7. cc_import(
  8.    name = "cupti_shared_library",
  9.    hdrs = [":headers"],
  10.    shared_library = "lib/libcupti.so.%{libcupti_version}",
  11. )
  12. '''
  13. cc_library(
  14.     name = "cupti",
  15.     #deps = [":cupti_shared_library"],
  16.     visibility = ["//visibility:public"],
  17. )
  18.  
  19. cc_library(
  20.     name = "headers",
  21.     #hdrs = glob([
  22.         #"include/Openacc/**",
  23.         #"include/Openmp/**",
  24.         #"include/cuda_stdint.h",
  25.         #"include/cupti.h",
  26.         #"include/cupti_activity.h",
  27.         #"include/cupti_activity_deprecated.h",
  28.         #"include/cupti_callbacks.h",
  29.         #"include/cupti_checkpoint.h",
  30.         #"include/cupti_driver_cbid.h",
  31.         #"include/cupti_events.h",
  32.         #"include/cupti_metrics.h",
  33.         #"include/cupti_nvtx_cbid.h",
  34.         #"include/cupti_pcsampling.h",
  35.         #"include/cupti_pcsampling_util.h",
  36.         #"include/cupti_profiler_target.h",
  37.         #"include/cupti_result.h",
  38.         #"include/cupti_runtime_cbid.h",
  39.         #"include/cupti_sass_metrics.h",
  40.         #"include/cupti_target.h",
  41.         #"include/cupti_version.h",
  42.         #"include/generated_cudaGL_meta.h",
  43.         #"include/generated_cudaVDPAU_meta.h",
  44.         #"include/generated_cuda_gl_interop_meta.h",
  45.         #"include/generated_cuda_meta.h",
  46.         #"include/generated_cuda_runtime_api_meta.h",
  47.         #"include/generated_cuda_vdpau_interop_meta.h",
  48.         #"include/generated_cudart_removed_meta.h",
  49.         #"include/generated_nvtx_meta.h",
  50.         #"include/nvperf_common.h",
  51.         #"include/nvperf_cuda_host.h",
  52.         #"include/nvperf_host.h",
  53.         #"include/nvperf_target.h",
  54.     #]) + if_version_equal_or_greater_than(
  55.         #"%{libcupti_minor_version}",
  56.         #"2024.0",
  57.         #["include/cupti_common.h"],
  58.     #),
  59.     include_prefix = "third_party/gpus/cuda/extras/CUPTI/include",
  60.     includes = ["include/"],
  61.     strip_include_prefix = "include",
  62.     visibility = ["@local_config_cuda//cuda:__pkg__"],
  63. )
  64.  
Tags: Jax
Advertisement
Add Comment
Please, Sign In to add comment