Relax Python 模块设计#
- Relax Python 模块设计概述
BasePyModule- PyTorch 和 TVM 之间的 DLPack 集成
- 测试从 PyTorch 张量到 TVM NDArray 的 DLPack 转换
- 测试从 GPU 上的 PyTorch 张量到 TVM NDArray 的 DLPack 转换
- 测试从 TVM NDArray 到 PyTorch 张量的 DLPack 转换
- 测试从 GPU 上的 TVM NDArray 到 PyTorch 张量的 DLPack 转换
- 测试 DLPack 往返转换:PyTorch -> TVM -> PyTorch
- 测试不同数据类型的 DLPack 转换
- 测试不同形状张量的 DLPack 转换
- 测试 DLPack 和 numpy 转换是否产生相同的结果
- 测试 DLPack 对不支持操作的错误处理
- 测试在 BasePyModule 上下文中的 DLPack 转换
- 测试 DLPack 转换保持设备一致性
- 测试 DLPack 转换在可能的情况下共享内存
- 测试批处理操作中的 DLPack 转换
- 测试 DLPack 在边缘情况下的转换
- PyTorch 与 TVM Relax 的集成
- TVMScript
@I.pyfunc装饰器 - 直方图
- 测试