Commit dc5bdb6b by tqchen Committed by Tianqi Chen

Upgrade TVM to latest version

parent 75ea6e45
...@@ -58,7 +58,7 @@ VTA_LIB_SRC = $(wildcard src/*.cc src/tvm/*.cc) ...@@ -58,7 +58,7 @@ VTA_LIB_SRC = $(wildcard src/*.cc src/tvm/*.cc)
ifeq ($(TARGET), VTA_PYNQ_TARGET) ifeq ($(TARGET), VTA_PYNQ_TARGET)
VTA_LIB_SRC += $(wildcard src/pynq/*.cc) VTA_LIB_SRC += $(wildcard src/pynq/*.cc)
LDFLAGS += -L/usr/lib -lsds_lib LDFLAGS += -L/usr/lib -lsds_lib
LDFLAGS += -L/opt/python3.6/lib/python3.6/site-packages/pynq/lib/ -l:libdma.so LDFLAGS += -L/opt/python3.6/lib/python3.6/site-packages/pynq/drivers/ -l:libdma.so
endif endif
VTA_LIB_OBJ = $(patsubst %.cc, build/%.o, $(VTA_LIB_SRC)) VTA_LIB_OBJ = $(patsubst %.cc, build/%.o, $(VTA_LIB_SRC))
......
#!/bin/bash #!/bin/bash
export PYTHONPATH=${PYTHONPATH}:/home/xilinx/tvm/python export PYTHONPATH=${PYTHONPATH}:/home/xilinx/tvm/python
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/opt/python3.6/lib/python3.6/site-packages/pynq/lib/ export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/opt/python3.6/lib/python3.6/site-packages/pynq/drivers/
python -m tvm.exec.rpc_server --load-library /home/xilinx/vta/lib/libvta.so python -m tvm.exec.rpc_server --load-library /home/xilinx/vta/lib/libvta.so
...@@ -129,7 +129,6 @@ with nnvm.compiler.build_config(opt_level=3): ...@@ -129,7 +129,6 @@ with nnvm.compiler.build_config(opt_level=3):
sym, target, shape_dict, dtype_dict, sym, target, shape_dict, dtype_dict,
params=params) params=params)
remote = rpc.connect(host, port)
temp = util.tempdir() temp = util.tempdir()
lib.save(temp.relpath("graphlib.o")) lib.save(temp.relpath("graphlib.o"))
remote.upload(temp.relpath("graphlib.o")) remote.upload(temp.relpath("graphlib.o"))
......
...@@ -52,4 +52,4 @@ def debug_mode(debug_flag): ...@@ -52,4 +52,4 @@ def debug_mode(debug_flag):
# Add a lower pass to sync uop # Add a lower pass to sync uop
build_module.BuildConfig.current.add_lower_pass = debug_mode(0) build_module.current_build_config().add_lower_pass = debug_mode(0)
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