PyPTO-Gym Tests
本目录是 pypto-gym 的测试目录,与 src/pypto_gym/ops/pypto_tensor/ 下的算子实现一一对应,例如:
运行方式
# 运行指定模型的所有算子
pytest tests/ops/arctic -v --forked
# 运行单个算子用例文件
pytest tests/ops/glm_v4_5/test_glm_gate.py -v
# 指定 NPU 设备
pytest tests/ops/arctic -v --forked --device 0
# 运行 experimental 目录下的算子(需显式指定路径)
pytest src/pypto_gym/ops/experimental/matmul -v --forked
标记(markers)
@pytest.mark.soc("950", "910")— 指定 SoC 版本。未标注等同于"910"。@pytest.mark.world_size(N)— 指定所需 NPU 卡数,默认 1。
SoC 筛选、耗时估计重排序等调度逻辑定义在顶层的 conftest.py。