[incubator-mxnet] 10/11: cython timing tests

[cj...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

cjolivier01 pushed a commit to branch cython
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git

commit 60dd961e9404bde52eea2b8d3fe1a5cf90aea3de
Author: Olivier <co...@amazon.com>
AuthorDate: Fri Mar 9 13:51:43 2018 -0800

    cython timing tests
---
 CMakeLists.txt                       | 2 ++
 tests/python/unittest/test_cython.py | 7 ++++---
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 04f0795..6e551f1 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -93,6 +93,8 @@ if("$ENV{VERBOSE}" STREQUAL "1")
   set(CMAKE_VERBOSE_MAKEFILE ON)
 endif()
 
+# Include dir for CLion IDE only, don't care which version
+find_package(PythonLibs)
 
 if(MSVC)
   add_definitions(-DWIN32_LEAN_AND_MEAN)
diff --git a/tests/python/unittest/test_cython.py b/tests/python/unittest/test_cython.py
index 0e4959e..5cb6829 100644
--- a/tests/python/unittest/test_cython.py
+++ b/tests/python/unittest/test_cython.py
@@ -102,8 +102,9 @@ if __name__ == '__main__':
   mxc.test_perf(iter_count, 0)
   test_perf(iter_count, 1)
   mxc.test_perf(iter_count, 1)
-  test_perf_bridge(iter_count, 0, 0)
-  test_perf_bridge(iter_count, 1, 0)
-  test_perf_bridge(iter_count, 1, 10)
+  test_perf_bridge(iter_count, 0, 0)  # Baseline, no cython call
+  test_perf_bridge(iter_count, 1, 0)  # Cython call, but no C API call
+  test_perf_bridge(iter_count, 1, 1)  # Cython call with one C API call
+  test_perf_bridge(iter_count, 1, 10) # Cython call with multiple C API call
 
 #  test_basic_cython()

-- 
To stop receiving notification emails like this one, please contact
cjolivier01@apache.org.