Commit 510105fc by ziheng Committed by Tianqi Chen

[FIX] Update rasp benchmark example (#225)

parent d7cc78a6
......@@ -23,7 +23,6 @@ def main():
args = parser.parse_args()
opt_level = args.opt_level
target = "llvm -target=armv7l-none-linux-gnueabihf -mcpu=cortex-a53 -mattr=+neon"
num_iter = args.num_iter
batch_size = 1
......@@ -43,9 +42,8 @@ def main():
with nnvm.compiler.build_config(opt_level=opt_level):
with tvm.target.rasp():
graph, lib, params = nnvm.compiler.build(
net, target, shape={"data": data_shape}, params=params)
graph, lib, params = nnvm.compiler.build(
net, tvm.target.rasp(), shape={"data": data_shape}, params=params)
tmp = util.tempdir()
lib_fname = tmp.relpath('net.o')
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment