Merge branch 'emma/fix_build_error_mlperf' into 'main'

fix build error mlperf See merge request dl/hugectr/hugectr!1487
NVIDIA-Merlin · Oct 20, 2023 · a925a88 · a925a88
2 parents 6171060 + b7779ce
commit a925a88
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/HugeCTR/src/layers/relu_layer.cu b/HugeCTR/src/layers/relu_layer.cu
@@ -36,7 +36,7 @@ template <typename MainOp, typename Fallback>
 __global__ void half4_relu_kernel(__half* __restrict__ out, const __half* __restrict__ in, int size,
                                   MainOp main_op, Fallback fallback) {
   const __half2 zero2 = TypeFunc<__half2>::zero();
-  const half4 zero4 = {zero2, zero2};
+  const half4 zero4 = half4{zero2, zero2};
   half4* out4 = reinterpret_cast<half4*>(out);
   const half4* in4 = reinterpret_cast<const half4*>(in);
 

diff --git a/test/utest/CMakeLists.txt b/test/utest/CMakeLists.txt
@@ -23,8 +23,10 @@ add_subdirectory(metrics)
 add_subdirectory(network)
 add_subdirectory(regularizers)
 add_subdirectory(device_map)
-add_subdirectory(embedding)
-add_subdirectory(embedding_training_cache)
+if (NOT DISABLE_CUDF)
+  add_subdirectory(embedding)
+  add_subdirectory(embedding_training_cache)
+endif()
 add_subdirectory(checker)
 add_subdirectory(prims)
 add_subdirectory(misc)
@@ -34,4 +36,4 @@ add_subdirectory(io)
 add_subdirectory(communication)
 add_subdirectory(pipeline)
 add_subdirectory(core23)
-add_subdirectory(dense_layer)
+add_subdirectory(dense_layer)