Puppet Class: profile::amd_gpu

Defined in:
modules/profile/manifests/amd_gpu.pp

Overview

Class profile::amd_gpu

Parameters:

  • rocm_version (Any) (defaults to: lookup('profile::amd_gpu::rocm_version', { 'default_value' => undef }))
  • kfd_access_group (Any) (defaults to: lookup('profile::amd_gpu::kfd_access_group', { 'default_value' => undef }))


3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
# File 'modules/profile/manifests/amd_gpu.pp', line 3

class profile::amd_gpu (
    $rocm_version = lookup('profile::amd_gpu::rocm_version', { 'default_value' => undef }),
    $kfd_access_group = lookup('profile::amd_gpu::kfd_access_group', { 'default_value' => undef }),
) {

    if $rocm_version {
        $rocm_smi_path = '/opt/rocm/bin/rocm-smi'

        # Some ROCm packages from 3.8+ ship with libpython3.8 requirements,
        # so for the moment we explicitly deploy Python 3.8 on Buster.
        # https://phabricator.wikimedia.org/T275896
        require profile::python38

        class { 'amd_rocm':
            version          => $rocm_version,
            kfd_access_group => $kfd_access_group,
        }

        class { 'prometheus::node_amd_rocm':
            rocm_smi_path => $rocm_smi_path,
        }
    }
}