flexloopy.config.env

flexloopy.config.env#

环境配置模块

该模块用于在本地或CI环境中统一设置与 TVM、VTA 以及第三方工具链相关的 Python 路径与环境变量，确保调用端能够顺利 import 对应包并运行示例或训练/编译流程。

使用说明：

  • 在项目启动脚本或交互式会话中调用 set_tvm 指定 TVM 源码根目录；必要时调用 set_caffeproto 指定 CaffeProto 根目录。

  • 本模块仅添加路径和环境变量，不会改变 TVM/VTA 的功能实现。

示例：

>>> from flexloopy.config.env import set_tvm, set_caffeproto
>>> set_tvm("/path/to/tvm")
>>> set_caffeproto("/path/to/CaffeProto")

小技巧

路径分隔符在不同操作系统上不同，Windows 为 ;，Unix/Linux 为 :。

本模块对 sys.path 的处理跨平台，PYTHONPATH 的格式请按实际平台调整。

Functions#

set_tvm(→ None)

配置 TVM/VTA 的 Python 依赖路径与环境变量

set_caffeproto(→ None)

配置 CaffeProto 的 Python 依赖路径(仅用于测试，暂未在项目中使用)

Module Contents#

flexloopy.config.env.set_tvm(tvm_root: str | pathlib.Path) → None[源代码]#

配置 TVM/VTA 的 Python 依赖路径与环境变量

参数:

tvm_root (str | Path) -- TVM 源码工程的根目录路径，可以是 str 或 pathlib.Path。应指向包含 python/ 与 vta/python/ 子目录的仓库根。

返回:

函数会原地更新 sys.path 和部分 os.environ 项以便后续可直接 import tvm 与 import vta。

返回类型:

None

使用示例：
>>> from flexloopy.config.env import set_tvm
>>> set_tvm("D:/dev/tvm")
>>> import tvm, vta  # 成功导入即表示环境设置生效
flexloopy.config.env.set_caffeproto(caffeproto_root: str | pathlib.Path = '../../../tests/caffeproto') → None[源代码]#

配置 CaffeProto 的 Python 依赖路径(仅用于测试，暂未在项目中使用)

参数:

caffeproto_root (str | Path, optional) -- CaffeProto 工程根目录路径，默认相对路径 ../../../tests/caffeproto。应包含 python/ 子目录。

返回:

函数会在 sys.path 中追加 caffeproto_root/python，避免重复添加。

返回类型:

None

使用示例：
>>> from flexloopy.config.env import set_caffeproto
>>> set_caffeproto("D:/dev/CaffeProto")
>>> import caffe  # 如项目提供对应 Python 包
目录