快速上手#
安装与构建
安装依赖:
pip install -e .或pip install tvm-book如需从源码构建 FFI:设置
TVM_FFI_EXT_FROM_SOURCE=ON并使用pip install -e .
加载共享库
默认自动定位构建产物;可通过环境变量覆盖:
TVM_BOOK_LIB_PATH指定完整库路径,如d:/build/tvm_book.dllTVM_BOOK_LIB_DIR指定目录,自动在其中寻找库文件
基本用例
add_oneimport numpy as np import tvm_book x = np.array([1, 2, 3], dtype=np.float32) y = np.empty_like(x) tvm_book.add_one(x, y) assert np.allclose(y, x + 1)
自定义对象
import tvm_book pair = tvm_book.IntPair(3, 5) assert pair.a == 3 and pair.b == 5 assert pair.get_first() == 3
错误排查
未找到库:根据异常信息检查候选目录,或设置
TVM_BOOK_LIB_PATH/TVM_BOOK_LIB_DIRVTA 未链接:相关 API 调用将抛出错误,需在构建时链接 VTA 运行时