Relax#
- FNormalize
- 测试在解析 TVMScript 时是否抑制应用 FNormalize
- 测试在 C++ 的 ExprMutator 子类中是否应用 FNormalize
- 测试在 Python 的 ExprMutator 子类中是否应用 FNormalize
- 测试如果 FNormalize 不应用更改，IR 是否格式良好
- 测试如果 FNormalize 应用更改，IR 是否格式不良
- 测试 FNormalize 是否为 R.call_tir 内联参数元组
- 测试 FNormalize 是否为 R.call_tir 内联参数元组
- 测试 FNormalize 是否为 R.call_tir_inplace 内联参数元组
- 测试 FNormalize 是否为 R.call_tir_with_grad 内联参数元组
- 规范化处理
- 测试嵌套序列表达式的规范化处理
- 测试包含数据流块的嵌套序列表达式的规范化处理
- 测试深层嵌套序列表达式的规范化处理
- 测试在数据流块中嵌套非数据流块的错误情况
- 测试移除void类型变量的使用
- 布局变换
- 优化布局变换
- RemoveWeightLayoutRewriteBlock
- 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装饰器 - 直方图
- 测试
- 构建块
- 算子注册
tvm.relax.testingtvm.relax.frontend.nn- Relax ONNX 前端
- PyTorch ONNX Relax
- Caffe Relax 前端 (TODO)
- Relax 量化/反量化算子模块
- 测试自定义算子合法化变换功能
- 合法化 qdq
- 测试从 float32 到 int8 的量化算子的合法化变换
- 测试从 float16 到 uint8 的量化算子的合法化变换
- 测试符号形状输入下，从float32到int8的量化算子的合法化变换
- 测试使用标量参数（而非张量）时，从float32到int8的量化算子的合法化变换
- 测试使用一维常量数组作为参数时，从float32到int8的量化算子的合法化变换
- 测试使用标量参数时，从float16到int8的量化算子的合法化变换
- 测试从int8到float32的反量化算子的合法化变换
- 测试使用标量参数时，从int8到float32的反量化算子的合法化变换
- 测试符号形状输入下，从int8到float32的反量化算子的合法化变换
- 测试从int8到float16的反量化算子的合法化变换
- 测试使用标量参数时，从int8到float16的反量化算子的合法化变换
- 量化神经网络
- 测试 LambdaLift
- Relax VM 中带有内存作用域(scope)的朴素内存分配器