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.1a11.tar.gz (11.7 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.1a11-py3-none-any.whl (15.9 kB view details)

Uploaded Python 3

File details

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

File metadata

File hashes

Hashes for ms_performance_prechecker-0.0.1a11.tar.gz
Algorithm Hash digest
SHA256 60ec592acce74f0609be0766c5e9727887a0094afac34302fc697e0e196c379b
MD5 0fee077050d386da48cefc2bec962ad1
BLAKE2b-256 ffe90727e5287ba65d8f939379f882bd0e2a8a56155e52d054edaf63359d3d4c

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for ms_performance_prechecker-0.0.1a11-py3-none-any.whl
Algorithm Hash digest
SHA256 e372ccf3f7764ef5304a78bfc7037afc681ecdc9f40bb05995d3cf2446f470dd
MD5 3893dbceba90c62cc52a264ea3000071
BLAKE2b-256 e9f1eaa76de39372e3ef4201f57ce9baeb5d1b36beb27a6ddd3836a07ff5ae91

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