Puppet Class: profile::amd_gpu

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

Overview

SPDX-License-Identifier: Apache-2.0

Class profile::amd_gpu

Parameters:

  • rocm_version (Optional[String]) (defaults to: lookup('profile::amd_gpu::rocm_version', { 'default_value' => undef }))
  • allow_gpu_broader_access (Boolean) (defaults to: lookup('profile::amd_gpu::allow_gpu_broader_access', { 'default_value' => false }))
  • is_kubernetes_node (Boolean) (defaults to: lookup('profile::amd_gpu::is_kubernetes_node', { 'default_value' => false })) 


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

class profile::amd_gpu (
    Optional[String] $rocm_version = lookup('profile::amd_gpu::rocm_version', { 'default_value' => undef }),
    Boolean $allow_gpu_broader_access = lookup('profile::amd_gpu::allow_gpu_broader_access', { 'default_value' => false }),
    Boolean $is_kubernetes_node = lookup('profile::amd_gpu::is_kubernetes_node', { 'default_value' => false }),
) {

    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,
            allow_gpu_broader_access => $allow_gpu_broader_access,
            is_kubernetes_node       => $is_kubernetes_node,
        }

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