example_simple -> example_tensordot

Author: nicolasvasilache

README.md

@@ -29,7 +29,7 @@ After a few generations of `autotuning` on a 2-GPU P100 system, we see results r
 
 ![Autotuning Sample](docs/source/_static/img/autotuning.png)
 
-In C++ a minimal autotuning example resembles the [following](example/example_simple.cc):
+In C++ a minimal autotuning example resembles the [following](example/example_tensordot.cc):
 ```cpp
 TEST(TensorDot, SimpleAutotune) {
   // 1. Define and setup the TC compilation unit with CUDA memory

examples/CMakeLists.txt

@@ -22,20 +22,25 @@ set(GTEST_LIBS gtest gtest_main)
 ################################################################################
 # Simple Example
 ################################################################################
-add_executable(example_simple example_simple.cc)
-add_test(example_simple example_simple)
-target_link_libraries(
-   example_simple
+set(EXAMPLES_FILES
+  example_tensordot
+)
+foreach(i ${EXAMPLES_FILES})
+  add_executable(${i} ${i}.cc)
+  add_test(${i} ${i})
+  target_link_libraries(
+     ${i}
 
-   tc_autotuner
-   tc_core
+     tc_autotuner
+     tc_core
 
-   ${GTEST_LIBS}
-   ${GFLAGS_LIBRARIES}
-   ${GLOG_LIBRARIES}
+     ${GTEST_LIBS}
+     ${GFLAGS_LIBRARIES}
+     ${GLOG_LIBRARIES}
 
-   ${ATEN_LIBRARIES}
-)
+     ${ATEN_LIBRARIES}
+  )
+endforeach()
 
 ################################################################################
 # Examples

examples/example_simple.cc renamed to examples/example_tensordot.cc

@@ -36,6 +36,9 @@ DEFINE_uint32(pop_size, 10, "Population size to tune for");
 DEFINE_uint32(threads, 16, "Number of threads to tune with");
 DEFINE_string(gpus, "0", "List of gpus to evaluate on");
 
+// TODO: This temporary flag will be disabled later
+DEFINE_bool(disable_version_checks, false, "TODO: remove me");
+
 TEST(TensorDot, SimpleAutotune) {
   // 1. Define and setup the TC compilation unit with CUDA memory
   // management backed by ATen tensors.