tools/rules/cu.bzl

load("@rules_cuda//cuda:defs.bzl", "cuda_library")

NVCC_COPTS = ["--expt-relaxed-constexpr", "--expt-extended-lambda"]

def cu_library(name, srcs, copts = [], **kwargs):
    cuda_library(name, srcs = srcs, copts = NVCC_COPTS + copts, **kwargs)