Skip to content
Snippets Groups Projects
Commit 65434337 authored by Hashim Sharif's avatar Hashim Sharif
Browse files

Wrapping up Keras frontend automated test script

parent 91d9cd17
No related branches found
No related tags found
No related merge requests found
...@@ -69,7 +69,8 @@ class Benchmark: ...@@ -69,7 +69,8 @@ class Benchmark:
try: try:
subprocess.run([ subprocess.run([
"hpvm-clang", src_file, target_binary, "hpvm-clang", src_file, target_binary,
"-t", "tensor", "--conf-file", approx_conf_file "-t", "tensor", "--conf-file", approx_conf_file,
"-fno-exceptions"
], check=True) ], check=True)
except: except:
print ("\n\n ERROR: HPVM Compilation Failed!! \n\n") print ("\n\n ERROR: HPVM Compilation Failed!! \n\n")
......
...@@ -147,6 +147,7 @@ class BenchmarkTests: ...@@ -147,6 +147,7 @@ class BenchmarkTests:
print ("Failed: " + failed_test) print ("Failed: " + failed_test)
# Returns False if any of the tests failed
def printHPVMSummary(self): def printHPVMSummary(self):
failed_test_count = len(self.failed_hpvm_tests) failed_test_count = len(self.failed_hpvm_tests)
...@@ -161,9 +162,14 @@ class BenchmarkTests: ...@@ -161,9 +162,14 @@ class BenchmarkTests:
print ("****** Failed Tests *** \n") print ("****** Failed Tests *** \n")
for failed_test in self.failed_hpvm_tests: for failed_test in self.failed_hpvm_tests:
print ("Failed: " + failed_test) print ("Failed: " + failed_test)
if failed_test_count > 0:
return False
return True
if __name__ == "__main__": if __name__ == "__main__":
...@@ -191,22 +197,25 @@ if __name__ == "__main__": ...@@ -191,22 +197,25 @@ if __name__ == "__main__":
VGG16_ImageNet = Benchmark(CUR_SRC_PATH + "/vgg16_imagenet.py", "src/vgg16_imagenet_src", 69.46) VGG16_ImageNet = Benchmark(CUR_SRC_PATH + "/vgg16_imagenet.py", "src/vgg16_imagenet_src", 69.46)
#testMgr.addBenchmark(AlexNet) #testMgr.addBenchmark(AlexNet)
#testMgr.addBenchmark(AlexNet_ImageNet)
#testMgr.addBenchmark(AlexNet2) #testMgr.addBenchmark(AlexNet2)
testMgr.addBenchmark(LeNet) #testMgr.addBenchmark(LeNet)
#testMgr.addBenchmark(MobileNet) testMgr.addBenchmark(MobileNet)
#testMgr.addBenchmark(ResNet18) testMgr.addBenchmark(ResNet18)
#testMgr.addBenchmark(ResNet50) #testMgr.addBenchmark(ResNet50)
#testMgr.addBenchmark(VGG16_cifar10) #testMgr.addBenchmark(VGG16_cifar10)
testMgr.addBenchmark(VGG16_cifar100) testMgr.addBenchmark(VGG16_cifar100)
#testMgr.addBenchmark(VGG16_ImageNet) #testMgr.addBenchmark(VGG16_ImageNet)
testMgr.addBenchmark(AlexNet_ImageNet)
#testMgr.runKerasTests() #testMgr.runKerasTests()
#testMgr.printKerasSummary() #testMgr.printKerasSummary()
testMgr.runHPVMTests() testMgr.runHPVMTests()
testMgr.printHPVMSummary() tests_passed = testMgr.printHPVMSummary()
if not tests_passed:
sys.exit(-1)
#testMgr.runKerasTests() #testMgr.runKerasTests()
#testMgr.printKerasSummary() #testMgr.printKerasSummary()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment