/ src / examples / nv / BUCK
BUCK
 1  # examples/nv/BUCK
 2  #
 3  # NVIDIA examples using clang (NOT nvcc)
 4  #
 5  # NOTE: Requires nvidia-sdk in the toolchain.
 6  
 7  load("@toolchains//:nv.bzl", "nv_binary")
 8  
 9  nv_binary(
10      name = "hello",
11      srcs = ["hello.cpp"],
12      visibility = ["PUBLIC"],
13  )
14  
15  nv_binary(
16      name = "mdspan_device_test",
17      srcs = ["mdspan_device_test.cpp"],
18      visibility = ["PUBLIC"],
19  )
20  
21  nv_binary(
22      name = "tensor_core",
23      srcs = ["tensor_core.cpp"],
24      visibility = ["PUBLIC"],
25  )