VX_NN - Link OpenCL Lib (#608)

ROCm · Aug 26, 2021 · 179948a · 179948a
1 parent a068211
commit 179948a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/amd_openvx_extensions/amd_nn/CMakeLists.txt b/amd_openvx_extensions/amd_nn/CMakeLists.txt
@@ -112,7 +112,7 @@ if(GPU_SUPPORT AND "${BACKEND}" STREQUAL "OPENCL"  AND OpenCL_FOUND)
     add_definitions(-DENABLE_OPENCL=${ENABLE_OPENCL} -DENABLE_HIP=${ENABLE_HIP})
     include_directories(${OpenCL_INCLUDE_DIRS} ${OpenCL_INCLUDE_DIRS}/Headers)
     add_library(vx_nn SHARED ${SOURCES})
-    target_link_libraries(vx_nn openvx MIOpen miopengemm)
+    target_link_libraries(vx_nn openvx MIOpen miopengemm ${OpenCL_LIBRARIES})
 elseif (GPU_SUPPORT AND "${BACKEND}" STREQUAL "HIP" AND HIP_FOUND)
     message("-- ${Green}amd_nn -- Building with HIP${ColourReset}")
     set(ENABLE_OPENCL 0)