15 lines
320 B
Plaintext
15 lines
320 B
Plaintext
|
load("@bazel_skylib//rules:copy_file.bzl", "copy_file")
|
||
|
|
|
||
|
|
copy_file(
|
||
|
|
name = "libpjrt_tpu_so",
|
||
|
|
src = "libtpu/libtpu.so",
|
||
|
|
out = "libpjrt_tpu.so",
|
||
|
|
allow_symlink = True,
|
||
|
|
)
|
||
|
|
|
||
|
|
cc_import(
|
||
|
|
name = "libpjrt_tpu",
|
||
|
|
shared_library = ":libpjrt_tpu_so",
|
||
|
|
visibility = ["@zml//runtimes/tpu:__subpackages__"],
|
||
|
|
)
|