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.1a16.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.1a16-py3-none-any.whl (37.2 kB view details)

Uploaded Python 3

File details

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

File metadata

File hashes

Hashes for ms_performance_prechecker-0.0.1a16.tar.gz
Algorithm Hash digest
SHA256 526f12130c323236ff1f541b89fe8bffc1ce850202bd2739b380556fe00ac28c
MD5 d070a2db709705b043b6072f431ac75f
BLAKE2b-256 47f430ee134c7c61de4098145ce1b9ce7e3355f5ac2525359d1c281e9993c5e3

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for ms_performance_prechecker-0.0.1a16-py3-none-any.whl
Algorithm Hash digest
SHA256 4efda16325cea7da1e093fed5a7a5e961115ea9efa15d922a88bba5e2fd60072
MD5 e1243005f20720c35c3f86e183439ab7
BLAKE2b-256 e8b4679f565737343b50e25e959d342c3796b89f39a72454546bb9f39bdd8528

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