fix cmake bug (apache#21)

tmoreau89 · Jan 2, 2019 · 7e40f05 · 7e40f05
1 parent 807bd51
commit 7e40f05
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cmake/modules/VTA.cmake b/cmake/modules/VTA.cmake
@@ -39,8 +39,10 @@ elseif(PYTHON)
   endif(APPLE)
 
   # PYNQ rules for pynq v2.3
-  find_library(__cma_lib NAMES cma PATH /usr/lib)
-  target_link_libraries(vta ${__cma_lib})
+  if (${VTA_TARGET} STREQUAL "pynq" OR ${VTA_TARGET} STREQUAL "ultra96")
+    find_library(__cma_lib NAMES cma PATH /usr/lib)
+    target_link_libraries(vta ${__cma_lib})
+  endif()
 else()
   message(STATUS "Cannot found python in env, VTA build is skipped..")
 endif()