Skip to main content

MindStudio Performance Prechecker

Project description

MindStudio Performance Prechecker

介绍

  • 基本功能 检查当前环境中 log level、cpu 绑核、内核版本等,是否达到最佳要求,并给出相应建议

  • 环境要求

    • Python >= 3.8
  • 安装

    pip install ms-performance-prechecker
    
  • 执行检查

    ms_performance_prechecker
    # ms_performance_prechecker_logger - INFO - <think>
    # ms_performance_prechecker_logger - INFO - simple_env_checker
    # ms_performance_prechecker_logger - INFO - linux_kernel_release_checker
    # ms_performance_prechecker_logger - INFO - Got kernel_release: 5.15.167.4-microsoft-standard-WSL2, suggested is 5.10
    # ...
    # ms_performance_prechecker_logger - INFO - </think>
    # ms_performance_prechecker_logger - INFO -
    # ms_performance_prechecker_logger - INFO - <answer>
    # ms_performance_prechecker_logger - INFO - [env] TASK_QUEUE_ENABLE
    # ms_performance_prechecker_logger - INFO - [action] export TASK_QUEUE_ENABLE=2
    # ms_performance_prechecker_logger - INFO - [reason] 配置task_queue 算子下发队列优化登记,可能导致运行中NPU内存峰值上升
    # ms_performance_prechecker_logger - INFO -
    # ms_performance_prechecker_logger - INFO - [env] HCCL_OP_EXPANSION_MODE
    # ms_performance_prechecker_logger - INFO - [action] export HCCL_OP_EXPANSION_MODE=AIV
    # ms_performance_prechecker_logger - INFO - [reason] 配置通信算法的编排展开位置,代表通信算法的编排展开位置在Device侧的AI Vector Core 计算单元(MindIE 2.0.T3 和 MindIE 2.0.T3.1 是能AIV会有崩溃风险,请不要设置它)
    # ...
    # ms_performance_prechecker_logger - INFO -
    # ms_performance_prechecker_logger - INFO - [system] CPU 可能不是高性能模式
    # ms_performance_prechecker_logger - INFO - [action] 开启 CPU 高性能模式:cpupower -c all frequency-set -g performance
    # ms_performance_prechecker_logger - INFO - [reason] 在相同时延约束下,TPS会有~3%的提升
    # ms_performance_prechecker_logger - INFO -
    # ms_performance_prechecker_logger - INFO - </answer>
    
  • 参数

    参数 说明
    -t, --check_type 检查项类型,可选值:basic, deepseek
    -s, --save_env 保存环境变量相关改动输出路径,默认值 ms_performance_prechecker_env.sh
    -l, --log_level 日志级别,可选值 debug, info, warning, error
  • Python 接口调用

    from ms_performance_prechecker import run_precheck, save_env_contents
    
    run_precheck(mindie_service_config)  # 执行检查
    save_env_contents("foo_env.sh")  # 保存环境变量相关改动到 foo_env.sh
    

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

ms_performance_prechecker-0.0.1a15.tar.gz (26.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

ms_performance_prechecker-0.0.1a15-py3-none-any.whl (37.2 kB view details)

Uploaded Python 3

File details

Details for the file ms_performance_prechecker-0.0.1a15.tar.gz.

File metadata

File hashes

Hashes for ms_performance_prechecker-0.0.1a15.tar.gz
Algorithm Hash digest
SHA256 b90ab36357efe4543f2ea4a2411d6174f08a37e962278a846f1ce1dc5ac46dfc
MD5 2f43a46a8b82a3ef7509710672f48d6c
BLAKE2b-256 679c085b346a48dd6c0752cf1019e165c2099a469e30187a71595576eb3fe188

See more details on using hashes here.

File details

Details for the file ms_performance_prechecker-0.0.1a15-py3-none-any.whl.

File metadata

File hashes

Hashes for ms_performance_prechecker-0.0.1a15-py3-none-any.whl
Algorithm Hash digest
SHA256 4199937dd1e2acea930fbc92d23d3de62a9241b96f0ccab4b2b9f2fcf1c1e76d
MD5 7ba72f223641ed72a81104f1c18baa66
BLAKE2b-256 dc420abcd898090f2b84850177801eb62ee8ef5eb9a75394e70e9aa429f65a0a

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page