Browse Source

Adding deployment of ROCm k8s-device-plugin

Signed-off-by: Luke Wilson <luke.wilson@dell.com>
Luke Wilson 4 years ago
parent
commit
6f699e5484
2 changed files with 9 additions and 1 deletions
  1. 6 0
      roles/k8s_start_services/tasks/main.yml
  2. 3 1
      roles/k8s_start_services/vars/main.yml

+ 6 - 0
roles/k8s_start_services/tasks/main.yml

@@ -133,3 +133,9 @@
   register: fpga_enable
   when: "'fpga-device-plugin' not in k8s_pods.stdout"
   tags: init
+
+- name: Deploy ROCm Device plugin
+  command: "kubectl create -f '{{ rocm_device_plugin_yaml_url }}'"
+  changed_when: true
+  when: ansible_local.inventory.amd_gpu > 0 
+  tags: init

+ 3 - 1
roles/k8s_start_services/vars/main.yml

@@ -43,4 +43,6 @@ mig_strategy: none
 
 gpu_feature_discovery_version: 0.2.0
 
-fpga_device_plugin_yaml_url: https://raw.githubusercontent.com/Xilinx/FPGA_as_a_Service/master/k8s-fpga-device-plugin/fpga-device-plugin.yml
+fpga_device_plugin_yaml_url: https://raw.githubusercontent.com/Xilinx/FPGA_as_a_Service/master/k8s-fpga-device-plugin/fpga-device-plugin.yml
+
+rocm_device_plugin_yaml_url: https://raw.githubusercontent.com/RadeonOpenCompute/k8s-device-plugin/master/k8s-ds-amdgpu-dp.yaml