Creating 2 sources for Sampling and Perforation Tests

hpvm/projects/hpvm-tensor-rt/CMakeLists.txt

@@ -140,8 +140,12 @@ add_tensor_runtime(tensor_runtime_online -DONLINE_PROFILING=true -DFP16_tuning=f
 add_dependencies(tensor_runtime_online tensor_runtime)
 
 # Adding rule for the debugging source
-add_executable(unit_tests tests/unit_tests.cc)
-target_link_libraries(unit_tests tensor_runtime_online)
+add_executable(sampling_tests tests/sampling_tests.cc)
+target_link_libraries(sampling_tests tensor_runtime_online)
+
+add_executable(perforation_tests tests/perforation_tests.cc)
+target_link_libraries(perforation_tests tensor_runtime_online)
+
 
 # -- Compile tensor_runtime.ll if possible
 if(INDEP_BUILD)

hpvm/projects/hpvm-tensor-rt/tests/perforation_tests.cc

+
+#include "tests.h"
+
+
+int main() {
+
+  llvm_hpvm_initTensorRt(0);
+
+  UnitTestResults unitTestResults;
+  
+  testPerforation(unitTestResults);
+  
+  unitTestResults.printSummary();
+
+  return 0;
+}

hpvm/projects/hpvm-tensor-rt/tests/sampling_tests.cc

+
+#include "tests.h"
+
+
+int main() {
+
+  llvm_hpvm_initTensorRt(0);
+
+  UnitTestResults unitTestResults;
+  
+  testSampling(unitTestResults); 
+  
+  unitTestResults.printSummary();
+
+  return 0;
+}

hpvm/projects/hpvm-tensor-rt/tests/unit_tests.cc → hpvm/projects/hpvm-tensor-rt/tests/tests.h

 
+
 #include <stdio.h>
 #include <stdlib.h>
 #include <unistd.h>
@@ -460,17 +461,9 @@ void testSampling_1_1(UnitTestResults &unitTestResults) {
 
 
 
-int main() {
-
-  llvm_hpvm_initTensorRt(0);
-
-  UnitTestResults unitTestResults;
+void testSampling(UnitTestResults &unitTestResults){
 
   testSampling_3_3(unitTestResults);
   testSampling_1_1(unitTestResults);
-  testPerforation(unitTestResults);
-  unitTestResults.printSummary();
-
-
-  return 0;
 }
+