summaryrefslogtreecommitdiff
path: root/compiler/tfts/CMakeLists.txt
blob: eda39b6ef7bf3d6d53ec7bf4ae016fdfa697b40f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
nncc_find_resource(TensorFlowTests)
nnas_find_package(TensorFlow QUIET)

if(NOT TensorFlow_FOUND)
  return()
endif(NOT TensorFlow_FOUND)

if(NOT TARGET tfkit)
  return()
endif(NOT TARGET tfkit)

message(STATUS "Build tfts: TRUE")

# NOTE
#
# tfts_check fails on UNIT_Const_001 and UNIT_Shape_000 due to
# the limitation of "nnkit_tf_backend"
#
# "nnkit_tf_backend" supports only TF_FLOAT32 input/output.
#
# TODO Remove nnkit dependency
add_custom_target(tfts_check
  COMMAND
    "${CMAKE_CURRENT_SOURCE_DIR}/check_all.sh"
    "${TensorFlowTests_DIR}"
    $<TARGET_FILE:tfkit>
    $<TARGET_FILE:nnkit-run>
    $<TARGET_FILE:nnkit_tf_backend>
  WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}
)