[Feat] add the API assign_values

- Add `check_evict_strategy` for `assign`
NVIDIA-Merlin · Nov 1, 2023 · 8a4b781 · 8a4b781
1 parent 5301c4a
commit 8a4b781
Show file tree

Hide file tree

Showing 5 changed files with 1,772 additions and 8 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -145,3 +145,8 @@ add_executable(accum_or_assign_test tests/accum_or_assign_test.cc)
 target_compile_features(accum_or_assign_test PUBLIC cxx_std_14)
 set_target_properties(accum_or_assign_test PROPERTIES  CUDA_ARCHITECTURES OFF)
 TARGET_LINK_LIBRARIES(accum_or_assign_test gtest_main)
+
+add_executable(assign_values_test tests/assign_values_test.cc.cu)
+target_compile_features(assign_values_test PUBLIC cxx_std_14)
+set_target_properties(assign_values_test PROPERTIES  CUDA_ARCHITECTURES OFF)
+TARGET_LINK_LIBRARIES(assign_values_test gtest_main)
diff --git a/include/merlin/core_kernels.cuh b/include/merlin/core_kernels.cuh
@@ -26,6 +26,7 @@
 #include "core_kernels/lookup_ptr.cuh"
 #include "core_kernels/update.cuh"
 #include "core_kernels/update_score.cuh"
+#include "core_kernels/update_values.cuh"
 #include "core_kernels/upsert.cuh"
 #include "core_kernels/upsert_and_evict.cuh"