diff options
Diffstat (limited to 'amd/packages/rocm-tools.scm')
| -rw-r--r-- | amd/packages/rocm-tools.scm | 131 |
1 files changed, 131 insertions, 0 deletions
diff --git a/amd/packages/rocm-tools.scm b/amd/packages/rocm-tools.scm new file mode 100644 index 0000000..2b58017 --- /dev/null +++ b/amd/packages/rocm-tools.scm @@ -0,0 +1,131 @@ +;;; Copyright © 2023 Advanced Micro Devices, Inc. +;;; +;;; This file is part of GNU Guix. +;;; +;;; GNU Guix is free software; you can redistribute it and/or modify it +;;; under the terms of the GNU General Public License as published by +;;; the Free Software Foundation; either version 3 of the License, or (at +;;; your option) any later version. +;;; +;;; GNU Guix is distributed in the hope that it will be useful, but +;;; WITHOUT ANY WARRANTY; without even the implied warranty of +;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +;;; GNU General Public License for more details. +;;; +;;; You should have received a copy of the GNU General Public License +;;; along with GNU Guix. If not, see <http://www.gnu.org/licenses/>. + +(define-module (amd packages rocm-tools) + #:use-module (guix packages) + #:use-module (guix gexp) + #:use-module (guix build-system cmake) + #:use-module (guix build-system python) + #:use-module (guix git-download) + #:use-module (guix licenses) + + #:use-module (gnu packages) + #:use-module (gnu packages base) + #:use-module (gnu packages linux) + #:use-module (gnu packages python) + #:use-module (gnu packages python-xyz) + #:use-module (gnu packages python-science) + #:use-module (gnu packages serialization) + + #:use-module (amd packages rocm-origin) + #:use-module (amd packages rocm-base)) + +; rocminfo +(define (make-rocminfo rocr-runtime) + (package + (name "rocminfo") + (version (package-version rocr-runtime)) + (source + (rocm-origin name version)) + (build-system cmake-build-system) + (arguments + `(#:build-type "Release" + #:tests? #f)) + (inputs (list rocr-runtime python)) + (propagated-inputs (list grep kmod)) + (synopsis "ROCm application for reporting system info") + (description + "List @acronym{HSA,Heterogeneous System Architecture} Agents available to ROCm and show their properties.") + (home-page "https://github.com/RadeonOpenCompute/rocminfo") + (license ncsa))) + +(define-public rocminfo + (make-rocminfo rocr-runtime)) + +; rocm-bandwidth-test +(define (make-rocm-bandwidth rocr-runtime) + (package + (name "rocm-bandwidth-test") + (version (package-version rocr-runtime)) + (source + (rocm-origin "rocm_bandwidth_test" version)) + (build-system cmake-build-system) + (arguments + (list + #:tests? #f ;No tests. + #:configure-flags #~(list "-DCMAKE_CXX_FLAGS=-Wno-error=cpp" ;ignores include reorg issue + ))) + (inputs (list rocr-runtime)) + (synopsis "Bandwith test for ROCm") + (description + "RocBandwidthTest is designed to capture the performance +characteristics of buffer copying and kernel read/write operations. The help +screen of the benchmark shows various options one can use in initiating +copy/read/writer operations. In addition one can also query the topology of the +system in terms of memory pools and their agents.") + (home-page "https://github.com/ROCm/rocm_bandwidth_test") + (license ncsa))) + +(define-public rocm-bandwidth + (make-rocm-bandwidth rocr-runtime)) + +; rocm-smi +(define (make-rocm-smi version) + (package + (name "rocm-smi") + (version version) + (source + (rocm-origin "rocm_smi_lib" version)) + (build-system cmake-build-system) + (arguments + `(#:build-type "Release" + #:tests? #f)) ;No tests. + (inputs (list python)) + (propagated-inputs (list grep coreutils)) + (synopsis "The ROCm System Management Interface (ROCm SMI) Library") + (description + "The ROCm System Management Interface Library, or ROCm SMI library, +is part of the Radeon Open Compute ROCm software stack. It is a C library for Linux that +provides a user space interface for applications to monitor and control GPU applications.") + (home-page "https://github.com/RadeonOpenCompute/rocm_smi_lib.git") + (license ncsa))) + +(define-public rocm-smi + (make-rocm-smi rocm-version-latest)) + +; tensile +(define (make-tensile version) + (package + (name "tensile") + (version version) + (source + (rocm-origin name version)) + (build-system python-build-system) + (native-inputs (list python-pandas)) + (propagated-inputs (list msgpack-3 python-msgpack python-pyyaml + python-joblib python-psutil)) + (synopsis "A GEMM kernel generator for AMD GPUs.") + (description + "Tensile is a tool for creating benchmark-driven backend libraries for GEMMs, GEMM-like problems +(such as batched GEMM), and general N-dimensional tensor contractions on a GPU. The Tensile library +is mainly used as backend library to rocBLAS. Tensile acts as the performance backbone for a wide +variety of 'compute' applications running on AMD GPUs.") + (home-page "https://github.com/ROCmSoftwarePlatform/Tensile.git") + (license expat))) + +(define-public tensile + (make-tensile rocm-version-latest)) |