mlx-benchmark 项目使用教程

mlx-benchmark 项目使用教程

mlx-benchmarkBenchmark of Apple's MLX operations on all Apple Silicon (GPU, CPU) + MPS and CUDA.项目地址:https://gitcode.com/gh_mirrors/ml/mlx-benchmark

1. 项目的目录结构及介绍

mlx-benchmark/
├── benchmarks/
│   ├── average_benchmark.md
│   └── ...
├── images/
│   └── ...
├── mlx_benchmark/
│   └── ...
├── .gitignore
├── .pre-commit-config.yaml
├── CONTRIBUTING.md
├── LICENSE
├── README.md
├── requirements.txt
└── ...

• benchmarks/: 包含各种基准测试文件。
• images/: 存放项目相关的图片资源。
• mlx_benchmark/: 项目的主要代码目录。
• .gitignore: Git 忽略文件配置。
• .pre-commit-config.yaml: 预提交钩子配置文件。
• CONTRIBUTING.md: 贡献指南。
• LICENSE: 项目许可证。
• README.md: 项目介绍和使用说明。
• requirements.txt: 项目依赖文件。

2. 项目的启动文件介绍

项目的启动文件是 run_benchmark.py，位于 mlx_benchmark 目录下。该文件用于运行基准测试，可以通过命令行参数配置不同的运行选项。

python run_benchmark.py --include_mps=False --include_cpu=False --include_mlx_cpu=False

3. 项目的配置文件介绍

项目的配置文件包括 .pre-commit-config.yaml 和 requirements.txt。

• .pre-commit-config.yaml: 用于配置预提交钩子，确保代码提交前通过一系列检查。
• requirements.txt: 列出了项目运行所需的所有依赖包及其版本。

# .pre-commit-config.yaml 示例
repos:
  - repo: https://github.com/pre-commit/pre-commit-hooks
    rev: v4.1.0
    hooks:
      - id: trailing-whitespace
      - id: end-of-file-fixer
      - id: check-yaml
      - id: check-added-large-files

# requirements.txt 示例
numpy==1.21.2
torch==1.9.0

通过以上配置文件，可以确保项目的代码质量和运行环境的稳定性。

mlx-benchmarkBenchmark of Apple's MLX operations on all Apple Silicon (GPU, CPU) + MPS and CUDA.项目地址:https://gitcode.com/gh_mirrors/ml/mlx-benchmark