Add a sample program to evaluate clock cycle profiler and support clock cycle profiling on android 64bit

Change: 149704577

2 files changed, 116 insertions, 0 deletions

diff --git a/tensorflow/contrib/hvx/clock_cycle_profiling/BUILD b/tensorflow/contrib/hvx/clock_cycle_profiling/BUILD
new file mode 100644
index 0000000000..621466c9ed
--- /dev/null
+++ b/tensorflow/contrib/hvx/clock_cycle_profiling/BUILD
@@ -0,0 +1,63 @@
+# Description:
+#   contains parts of TensorFlow that are experimental or unstable and which are not supported.
+
+licenses(["notice"])  # Apache 2.0
+
+load(
+    "//tensorflow:tensorflow.bzl",
+    "tf_copts",
+)
+
+exports_files(["LICENSE"])
+
+package(default_visibility = ["//visibility:public"])
+
+filegroup(
+    name = "all_files",
+    srcs = glob(
+        ["**/*"],
+        exclude = [
+            "**/METADATA",
+            "**/OWNERS",
+        ],
+    ),
+    visibility = ["//tensorflow:__subpackages__"],
+)
+
+cc_binary(
+    name = "clock_cycle_profiling",
+    testonly = 1,
+    srcs = ["clock_cycle_profiling_main.cc"],
+    copts = tf_copts(),
+    linkopts = select({
+        "//tensorflow:android": [
+            "-pie",
+            "-s",
+            "-landroid",
+            "-ljnigraphics",
+            "-llog",
+            "-lm",
+            "-z defs",
+            "-s",
+            "-Wl,--exclude-libs,ALL",  # Exclude syms in all libs from auto export
+        ],
+        "//conditions:default": [],
+    }),
+    linkstatic = 1,
+    visibility = ["//visibility:public"],
+    deps = select({
+        "//tensorflow:android": [
+            "//tensorflow/core:android_tensorflow_lib",
+            "//tensorflow/core:android_tensorflow_test_lib",
+        ],
+        "//conditions:default": [
+            "//tensorflow/core:core_cpu",
+            "//tensorflow/core:lib",
+            "//tensorflow/core:framework",
+            "//tensorflow/core:framework_internal",
+            "//tensorflow/core:protos_all_cc",
+            "//tensorflow/core:tensorflow",
+            "//tensorflow/core:test",
+        ],
+    }),
+)
diff --git a/tensorflow/contrib/hvx/clock_cycle_profiling/clock_cycle_profiling_main.cc b/tensorflow/contrib/hvx/clock_cycle_profiling/clock_cycle_profiling_main.cc
new file mode 100644
index 0000000000..a87ef953b7
--- /dev/null
+++ b/tensorflow/contrib/hvx/clock_cycle_profiling/clock_cycle_profiling_main.cc
@@ -0,0 +1,53 @@
+/* Copyright 2017 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+#include "tensorflow/core/platform/logging.h"
+#include "tensorflow/core/platform/profile_utils/clock_cycle_profiler.h"
+#include "tensorflow/core/platform/profile_utils/cpu_utils.h"
+
+int main(int argc, char** argv) {
+  static constexpr int LOOP_COUNT = 1000000;
+
+#if defined(__ANDROID_API__)
+#if defined(__aarch64__)
+  LOG(INFO) << "android arm 64 bit";
+#endif
+#if defined(__ARM_ARCH_7A__)
+  LOG(INFO) << "android arm 32 bit";
+#endif
+  LOG(INFO) << "Android API = " << __ANDROID_API__;
+  if (__ANDROID_API__ < 21) {
+    LOG(INFO) << "Cpu utils requires API level 21 or above.";
+    return 0;
+  }
+#endif
+
+  tensorflow::profile_utils::CpuUtils::EnableClockCycleProfiling(true);
+
+  tensorflow::ClockCycleProfiler prof_global;
+  tensorflow::ClockCycleProfiler prof_internal;
+
+  prof_global.Start();
+  for (int i = 0; i < LOOP_COUNT; ++i) {
+    prof_internal.Start();
+    prof_internal.Stop();
+  }
+  prof_global.Stop();
+
+  prof_global.DumpStatistics("prof_global");
+  prof_internal.DumpStatistics("prof_internal");
+
+  return 0;
+}