Save .so instead of .o in convolution_opt.py

Since the ext_dev target may be compiled separately.
apache · Jul 3, 2023 · 27153ac · 27153ac
1 parent 14b36ea
commit 27153ac
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/vta/tutorials/optimize/convolution_opt.py b/vta/tutorials/optimize/convolution_opt.py
@@ -374,9 +374,9 @@
         s, [data, kernel, res], tvm.target.Target("ext_dev", host=env.target_host), name="my_conv"
     )
 temp = utils.tempdir()
-my_conv.save(temp.relpath("conv2d.o"))
-remote.upload(temp.relpath("conv2d.o"))
-f = remote.load_module("conv2d.o")
+my_conv.export_library(temp.relpath("conv2d.so"))
+remote.upload(temp.relpath("conv2d.so"))
+f = remote.load_module("conv2d.so")
 
 # Get the remote device context
 ctx = remote.ext_dev(0)