EngineX-Cambricon/enginex-mlu370-vllm
10
0
Fork 1
Code Issues Pull Requests Actions Projects Releases Wiki Activity
Files
main
enginex-mlu370-vllm/torch_mlu_ops-v1.3.2/tests/ops_pytest
History
Chranos 79dfc69789 add ops
2026-02-04 17:39:32 +08:00
..
allreduce_case
add ops
2026-02-04 17:39:32 +08:00
common_utils.py
add ops
2026-02-04 17:39:32 +08:00
README.md
add ops
2026-02-04 17:39:32 +08:00
run_gen_case.py
add ops
2026-02-04 17:39:32 +08:00
run_test.sh
add ops
2026-02-04 17:39:32 +08:00
test_active.py
add ops
2026-02-04 17:39:32 +08:00
test_apply_rotary.py
add ops
2026-02-04 17:39:32 +08:00
test_attn_proj.py
add ops
2026-02-04 17:39:32 +08:00
test_batch_matmul.py
add ops
2026-02-04 17:39:32 +08:00
test_copy_blocks.py
add ops
2026-02-04 17:39:32 +08:00
test_dequant_from_linear_cache.py
add ops
2026-02-04 17:39:32 +08:00
test_dequant_from_paged_cache.py
add ops
2026-02-04 17:39:32 +08:00
test_faketensor.py
add ops
2026-02-04 17:39:32 +08:00
test_ffn.py
add ops
2026-02-04 17:39:32 +08:00
test_flash_attention.py
add ops
2026-02-04 17:39:32 +08:00
test_fused_attn_proj.py
add ops
2026-02-04 17:39:32 +08:00
test_fused_ffn.py
add ops
2026-02-04 17:39:32 +08:00
test_fused_layernorm.py
add ops
2026-02-04 17:39:32 +08:00
test_fused_rmsnorm.py
add ops
2026-02-04 17:39:32 +08:00
test_fused_rope.py
add ops
2026-02-04 17:39:32 +08:00
test_group_gemm.py
add ops
2026-02-04 17:39:32 +08:00
test_matmul.py
add ops
2026-02-04 17:39:32 +08:00
test_moe_add_bias_activation.py
add ops
2026-02-04 17:39:32 +08:00
test_moe_cast_gating.py
add ops
2026-02-04 17:39:32 +08:00
test_moe_combine_result.py
add ops
2026-02-04 17:39:32 +08:00
test_moe_expand_input.py
add ops
2026-02-04 17:39:32 +08:00
test_moe_gen_idx.py
add ops
2026-02-04 17:39:32 +08:00
test_moe_softmax_topk.py
add ops
2026-02-04 17:39:32 +08:00
test_moe.py
add ops
2026-02-04 17:39:32 +08:00
test_offline_quant_to_linear_cache.py
add ops
2026-02-04 17:39:32 +08:00
test_offline_quant_to_paged_cache.py
add ops
2026-02-04 17:39:32 +08:00
test_per_token_smooth_quantize.py
add ops
2026-02-04 17:39:32 +08:00
test_preload.py
add ops
2026-02-04 17:39:32 +08:00
test_quant_ffn.py
add ops
2026-02-04 17:39:32 +08:00
test_quant_to_linear_cache.py
add ops
2026-02-04 17:39:32 +08:00
test_quant_to_paged_cache.py
add ops
2026-02-04 17:39:32 +08:00
test_quantize.py
add ops
2026-02-04 17:39:32 +08:00
test_reshape_linear_cache.py
add ops
2026-02-04 17:39:32 +08:00
test_reshape_paged_cache.py
add ops
2026-02-04 17:39:32 +08:00
test_self_attn.py
add ops
2026-02-04 17:39:32 +08:00
test_session_cache_attn.py
add ops
2026-02-04 17:39:32 +08:00
test_single_query_cached_kv_attn.py
add ops
2026-02-04 17:39:32 +08:00
test_single_query_mixed_cached_kv_attn.py
add ops
2026-02-04 17:39:32 +08:00
test_smooth_quant_group_gemm.py
add ops
2026-02-04 17:39:32 +08:00
test_smooth_quant_matmul.py
add ops
2026-02-04 17:39:32 +08:00
test_smooth_quant.py
add ops
2026-02-04 17:39:32 +08:00
test_swap_blocks.py
add ops
2026-02-04 17:39:32 +08:00
test_update_out_and_lse.py
add ops
2026-02-04 17:39:32 +08:00
test_weight_only_quant_matmul.py
add ops
2026-02-04 17:39:32 +08:00

README.md

BT_OPS测试脚本使用方式

# 测试所有测例
bash run_test.sh

# 测试单个测例
python3 test_测例名称.py

  • 必须在Torch-MLU-Ops docker容器内运行。

  • 测试脚本的命名规则为 test_测例名称.py。

  • 必须保证 Torch-MLU-Ops whl包正确安装。