summaryrefslogtreecommitdiff
path: root/infra/cmake/packages/TensorFlowVersionChecker.c
diff options
context:
space:
mode:
Diffstat (limited to 'infra/cmake/packages/TensorFlowVersionChecker.c')
-rw-r--r--infra/cmake/packages/TensorFlowVersionChecker.c9
1 files changed, 9 insertions, 0 deletions
diff --git a/infra/cmake/packages/TensorFlowVersionChecker.c b/infra/cmake/packages/TensorFlowVersionChecker.c
new file mode 100644
index 000000000..6161ef74a
--- /dev/null
+++ b/infra/cmake/packages/TensorFlowVersionChecker.c
@@ -0,0 +1,9 @@
+#include <string.h>
+#include <tensorflow/c/c_api.h>
+
+int main(int argc, char **argv)
+{
+ if (argc >= 2 && !strcmp(argv[1], TF_Version()))
+ return 0;
+ return 255;
+}