33 lines
838 B
Plaintext
33 lines
838 B
Plaintext
load("@aspect_bazel_lib//lib:copy_to_directory.bzl", "copy_to_directory")
|
|
load("@zml//bazel:cc_import.bzl", "cc_import")
|
|
|
|
copy_to_directory(
|
|
name = "sandbox",
|
|
srcs = [
|
|
"@cuda_nvcc//:libdevice",
|
|
"@cuda_nvcc//:ptxas",
|
|
],
|
|
include_external_repositories = ["**"],
|
|
)
|
|
|
|
cc_import(
|
|
name = "libpjrt_cuda",
|
|
data = [":sandbox"],
|
|
shared_library = "libpjrt_cuda.so",
|
|
visibility = ["@zml//runtimes/cuda:__subpackages__"],
|
|
deps = [
|
|
"@cuda_cudart//:cudart",
|
|
"@cuda_cupti//:cupti",
|
|
"@cuda_nvcc//:nvvm",
|
|
"@cuda_nvrtc//:nvrtc",
|
|
"@cudnn//:cudnn",
|
|
"@libcublas//:cublas",
|
|
"@libcufft//:cufft",
|
|
"@libcusolver//:cusolver",
|
|
"@libcusparse//:cusparse",
|
|
"@libnccl//:nccl",
|
|
"@libnvjitlink//:nvjitlink",
|
|
"@zlib",
|
|
],
|
|
)
|