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)