ROCm/TheRock

The HIP Environment and ROCm Kit - A lightweight open source build system for HIP and ROCm

解決的問題

TheRock 簡化了 HIP 與 ROCm 的安裝與建構流程,消除傳統套件式安裝的複雜性。它提供一個輕量級的建構平台,讓開發者與研究人員能輕鬆存取最新的 ROCm 功能,包含 ROCm、PyTorch 與 JAX 的每日發布版本。

運作方式

它作為一個 CMake 超專案運作,負責管理 HIP 與 ROCm 的原始碼建構。使用者可透過功能旗標(例如啟用特定的數學或機器學習函式庫)自訂建構,並支援多種 Linux 發行版與原生 Windows。專案亦包含開發單一 ROCm 元件的工具,以及完整的 CI/CD 管線,用於測試與釋出。

適用對象

  • ROCm 貢獻者
  • 需要最新 ROCm 功能的開發者與研究人員
  • 希望避免套件式安裝複雜性的進階使用者

主要特色

  • ROCm、PyTorch 與 JAX 的每日發布版本
  • 支援從原始碼建構 PyTorch 與 JAX(基於 ROCm)
  • 支援多種 Linux 發行版與原生 Windows 的跨平台建構
  • 透過 CMake 標誌實現高度可自訂的建構設定
  • 集成 ccache 以加速頻繁的重新建構

相關

  • 專案
  • 專案
  • Dispatch
  • 專案
  • 專案