ROCm/legacy-rocm-build

AMD ROCm™ Software - GitHub Home

何を解決するか

ROCm は、AMD GPU を用いた高性能計算 (HPC)、AI、科学計算、コンピュータ支援設計 (CAD) に向けた GPU プログラミングを可能にするオープンソースのソフトウェアスタックを提供します。低レベルの GPU カーネルとエンドユーザー向けアプリケーションの間のギャップを埋め、複雑なワークロードに対する GPU 加速を実現します。

動作方法

ROCm は HIP によって駆動されており、NVIDIA CUDA に類似したインターフェースを持つ C++ ランタイム API およびカーネル言語です。開発者がポータブルな GPU コードを記述できるようにします。スタックには、以下の包括的なドライバ、開発ツール、API が含まれます:

  • 数学および計算ライブラリ: 行列計算、FFT、乱数生成に特化したライブラリ(例:rocBLAS、rocFFT)。
  • ランタイムおよびコンパイラ: LLVM や ROCr ランタイムなどの GPU ワークロードのビルドおよび実行に使用するツール。
  • 統合: PyTorch や TensorFlow などの主要な機械学習フレームワークとの直接互換性。
  • ツールキット: 専用のプロファイリング、デバッグ、モニタリングツール(例:ROCgdb、AMD SMI)。

対象ユーザー

AMD GPU ハードウェアを使用する AI 学習・推論、科学計算、高性能計算の分野で作業する開発者および研究者向けに設計されています。

特徴

  • CUDA に似たインターフェース: HIP を使用してポータブルな GPU プログラミングを可能にします。
  • ライブラリエコシステム: 数学、通信、メディア用の広範なライブラリを含みます。
  • 主要フレームワーク対応: PyTorch、TensorFlow、JAX と統合されています。
  • 広範なハードウェア対応: CDNA および RDNA アーキテクチャを、さまざまな Linux ディストリビューションでサポートしています。

関連

  • プロジェクト
  • プロジェクト
  • プロジェクト
  • プロジェクト
  • プロジェクト