Pytorch兼容性 (v1.4)
更新时间：
复制为 MD 格式
Pytorch 2.4 Unit Test测试
测试配置
Ubuntu 22.04
Pytorch 2.4.0
CUDA 12.3（PPU SDK 1.4 内部版本）
Python 3.10
测试结果
Test Case 总数 | 305,621 |
实际运行Test Case 总数 | 305,621 |
Pass 总数 | 228,293 |
Fail 总数 | 5,688 |
Skip 总数 | 71,641 |
Pass Rate | 97.6% |
说明
Pass Rate计算公式 =（Pass总数 / (实际运行总数-Skip总数)
Pytorch 2.4 Operator测试
测试配置
Ubuntu 22.04
Pytorch 2.4.0
CUDA 12.3（PPU SDK 1.4 内部版本）
Python 3.10
算子支持情况
Module | 包含的Operator数量 | PPU支持的Operator数量 |
torch | 476 | 476 |
torch.nn | 218 | 218 |
torch.nn.functional | 115 | 115 |
torch.cuda | 82 | 82 |
torch.Tensor | 526 | 526 |
torch.distributed | 48 | 48 |
torch.linalg | 41 | 41 |
torch.optim | 33 | 33 |
Total | 1,539 | 1,539 |
算子Unit Test结果
Test Case 总数 | 277,274 |
Pass 总数 | 207,396 |
Skip 总数 | 65,687 |
Fail 总数 | 3,847 |
Pass Rate | 98.0% |
说明
Pass Rate 计算公式 =（Pass总数 / (Test Case总数-Skip总数)
该文章对您有帮助吗？