Skip to content

Commit c1aa268

Browse files
committed
Upgrade TVM to latest version
1 parent bfc2367 commit c1aa268

File tree

4 files changed

+4
-5
lines changed

4 files changed

+4
-5
lines changed

vta/Makefile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ VTA_LIB_SRC = $(wildcard src/*.cc src/tvm/*.cc)
5858
ifeq ($(TARGET), VTA_PYNQ_TARGET)
5959
VTA_LIB_SRC += $(wildcard src/pynq/*.cc)
6060
LDFLAGS += -L/usr/lib -lsds_lib
61-
LDFLAGS += -L/opt/python3.6/lib/python3.6/site-packages/pynq/lib/ -l:libdma.so
61+
LDFLAGS += -L/opt/python3.6/lib/python3.6/site-packages/pynq/drivers/ -l:libdma.so
6262
endif
6363
VTA_LIB_OBJ = $(patsubst %.cc, build/%.o, $(VTA_LIB_SRC))
6464

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
#!/bin/bash
22
export PYTHONPATH=${PYTHONPATH}:/home/xilinx/tvm/python
3-
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/opt/python3.6/lib/python3.6/site-packages/pynq/lib/
3+
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/opt/python3.6/lib/python3.6/site-packages/pynq/drivers/
44
python -m tvm.exec.rpc_server --load-library /home/xilinx/vta/lib/libvta.so

vta/examples/resnet18/pynq/imagenet_predict.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
for file in [TEST_FILE, CATEG_FILE, RESNET_GRAPH_FILE, RESNET_PARAMS_FILE, BITSTREAM_FILE]:
3030
if not os.path.isfile(file):
3131
print ("Downloading {}".format(file))
32-
wget.download(url+file)
32+
wget.download(url+file)
3333

3434
# Program the FPGA remotely
3535
assert tvm.module.enabled("rpc")
@@ -129,7 +129,6 @@ def mark_nop(graph, conv_layer=-1, skip_conv_layer=()):
129129
sym, target, shape_dict, dtype_dict,
130130
params=params)
131131

132-
remote = rpc.connect(host, port)
133132
temp = util.tempdir()
134133
lib.save(temp.relpath("graphlib.o"))
135134
remote.upload(temp.relpath("graphlib.o"))

vta/python/vta/build.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,4 +52,4 @@ def add_debug(stmt):
5252

5353

5454
# Add a lower pass to sync uop
55-
build_module.BuildConfig.current.add_lower_pass = debug_mode(0)
55+
build_module.current_build_config().add_lower_pass = debug_mode(0)

0 commit comments

Comments
 (0)