diff --git a/benchmark/BUILD b/benchmark/BUILD
index a7221a33b..21f1d6f5a 100644
--- 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
index 7ab1304fa..949732cf7 100644
--- 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
index cbacc911e..d4255a4f1 100644
--- 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
index 16c29537b..03ca7ddcb 100644
--- 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
index 0593a0a6a..a836eb8b6 100644
--- 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([