Skip to content

Commit f20f78c

Browse files
committed
Save .so instead of .o in convolution_opt.py
Since the ext_dev target may be compiled separately.
1 parent 78a4840 commit f20f78c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

vta/tutorials/optimize/convolution_opt.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -374,9 +374,9 @@
374374
s, [data, kernel, res], tvm.target.Target("ext_dev", host=env.target_host), name="my_conv"
375375
)
376376
temp = utils.tempdir()
377-
my_conv.save(temp.relpath("conv2d.o"))
378-
remote.upload(temp.relpath("conv2d.o"))
379-
f = remote.load_module("conv2d.o")
377+
my_conv.export_library(temp.relpath("conv2d.so"))
378+
remote.upload(temp.relpath("conv2d.so"))
379+
f = remote.load_module("conv2d.so")
380380

381381
# Get the remote device context
382382
ctx = remote.ext_dev(0)

0 commit comments

Comments
 (0)