Skip to content

Commit bc76822

Browse files
Fix alexnet-tvm-vm smoke-test
1 parent 23c7a7b commit bc76822

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/smoke_test/benchmark_smoke/conftest.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ def convert_models_to_tvm(use_caffe: bool = False):
151151
f'-w {OUTPUT_DIR}/public/googlenet-v1/googlenet-v1.caffemodel '
152152
f'-op {OUTPUT_DIR}/public/googlenet-v1/ -f caffe')
153153
tvm_compiler = (f'cd {OUTPUT_DIR} && python3 {TVM_COMPILER} -m alexnet.json '
154-
'-p alexnet.params -t llvm --lib_name alexnet_vm.so -vm')
154+
'-p alexnet.params -t llvm --lib_name alexnet_vm.so -high_lever_api RelayVM')
155155

156156
if use_caffe:
157157
execute_process(command_line=caffe_to_tvm_converter, log=log)

0 commit comments

Comments
 (0)