fix cmake bug (apache#21)

jroesch · Dec 1, 2018 · e549c46 · e549c46
1 parent 8151627
commit e549c46
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()