1 files changed, 30 insertions, 0 deletions
diff --git a/compiler/tfts/CMakeLists.txt b/compiler/tfts/CMakeLists.txt
new file mode 100644
index 000000000..eda39b6ef
--- /dev/null
+++ b/compiler/tfts/CMakeLists.txt
@@ -0,0 +1,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}
+)