rename cuda defs

NVIDIA-Merlin · Apr 22, 2024 · d0da5c9 · d0da5c9
1 parent d42901f
commit d0da5c9
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 12 deletions.
diff --git a/benchmark/BUILD b/benchmark/BUILD
@@ -1,4 +1,4 @@
-load("@local_config_cuda//cuda:build_defs.bzl", "cuda_library")
+load("@local_config_cuda//cuda:build_defs.bzl", "cuda_cc_library")
 
 cc_binary(
     name = "benchmark_util",
@@ -7,7 +7,7 @@ cc_binary(
     ],
 )
 
-cuda_library(
+cuda_cc_library(
     name = "benchmark_lib",
     srcs = [
         "merlin_hashtable_benchmark.cc.cu",

diff --git a/build_deps/gpus/cuda/build_defs.bzl.tpl b/build_deps/gpus/cuda/build_defs.bzl.tpl
@@ -41,12 +41,12 @@ def cuda_header_library(name,
                       **kwargs)
 
 
-def cuda_library(copts=[], **kwargs):
+def cuda_cc_library(copts=[], **kwargs):
     """Wrapper over cc_library which adds default CUDA options."""
     native.cc_library(copts=cuda_default_copts() + copts, **kwargs)
 
 
-def cuda_binary(copts=[], **kwargs):
+def cuda_cc_binary(copts=[], **kwargs):
     """Wrapper over cc_library which adds default CUDA options."""
     native.cc_binary(copts=cuda_default_copts() + copts, **kwargs)
 

diff --git a/include/BUILD b/include/BUILD
@@ -1,6 +1,6 @@
-load("@local_config_cuda//cuda:build_defs.bzl", "cuda_library")
+load("@local_config_cuda//cuda:build_defs.bzl", "cuda_cc_library")
 
-cuda_library(
+cuda_cc_library(
     name = "merlin_localfile",
     hdrs = [
         "merlin_localfile.hpp",
@@ -14,7 +14,7 @@ cuda_library(
     ],
 )
 
-cuda_library(
+cuda_cc_library(
     name = "merlin_hashtable",
     hdrs = [
         "merlin_hashtable.cuh",

diff --git a/include/merlin/BUILD b/include/merlin/BUILD
@@ -1,6 +1,6 @@
-load("@local_config_cuda//cuda:build_defs.bzl", "cuda_library")
+load("@local_config_cuda//cuda:build_defs.bzl", "cuda_cc_library")
 
-cuda_library(
+cuda_cc_library(
     name = "types_and_utils",
     srcs = [
     ],
@@ -16,7 +16,7 @@ cuda_library(
     ],
 )
 
-cuda_library(
+cuda_cc_library(
     name = "merlin",
     srcs = [
     ],

diff --git a/include/merlin/core_kernels/BUILD b/include/merlin/core_kernels/BUILD
@@ -1,6 +1,6 @@
-load("@local_config_cuda//cuda:build_defs.bzl", "cuda_library")
+load("@local_config_cuda//cuda:build_defs.bzl", "cuda_cc_library")
 
-cuda_library(
+cuda_cc_library(
     name = "core_kernels",
     srcs = [],
     hdrs = glob([