From b7779ce5d880703554682dd192a39ba4bb6e684e Mon Sep 17 00:00:00 2001 From: Emma Qiao Date: Thu, 19 Oct 2023 23:13:28 -0700 Subject: [PATCH] fix build error mlperf --- HugeCTR/src/layers/relu_layer.cu | 2 +- test/utest/CMakeLists.txt | 8 +++++--- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/HugeCTR/src/layers/relu_layer.cu b/HugeCTR/src/layers/relu_layer.cu index 0d5b724ce3..bbddc160c7 100644 --- a/HugeCTR/src/layers/relu_layer.cu +++ b/HugeCTR/src/layers/relu_layer.cu @@ -36,7 +36,7 @@ template __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(out); const half4* in4 = reinterpret_cast(in); diff --git a/test/utest/CMakeLists.txt b/test/utest/CMakeLists.txt index df0aea2147..1043f87abc 100644 --- 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) \ No newline at end of file +add_subdirectory(dense_layer)