apiVersion: kustomize.config.k8s.io/v1beta1
kind: Kustomization

resources:
  - app.yaml
  - runtime-class.yaml

helmCharts:
  - name: nvidia-device-plugin
    repo: https://nvidia.github.io/k8s-device-plugin
    version: 0.17.0
    releaseName: nvidia-device-plugin
    namespace: gpu-system
    valuesFile: values.yaml
    includeCRDs: true
  - name: amd-gpu
    repo: https://rocm.github.io/k8s-device-plugin/
    version: 0.21.0
    releaseName: amd-gpu-device-plugin
    namespace: gpu-system
    valuesFile: amd-gpu-values.yaml
    includeCRDs: true

patches:
  - target:
      group: apps
      version: v1
      kind: DaemonSet
      name: amd-gpu-device-plugin-daemonset
      namespace: gpu-system
    patch: |-
      - op: replace
        path: /spec/template/spec/nodeSelector
        value:
          kubernetes.io/arch: amd64
          kubernetes.io/hostname: ai.tail2fe2d.ts.net