# proto-file: https://cos.googlesource.com/cos/tools/+/refs/heads/master/src/pkg/gpuconfig/proto/gpu_driver_versions.proto
# proto-message: GPUDriverVerionInfoList

# Description:
# This file contains GPU driver version and label information.
# It is used by GPU driver installer to install and list GPU drivers and to illustrate the labels associated with the drivers.

# Update Instruction:
# If the GPU driver will not supported by COS anymore, we need to delete them from all GPU types.
# One GPU driver may contain multiple labels for sepcific gpu type, and for each label, we need to create an entry for it.
# If the default/latest/R<major-version>/other GPU driver labels are updated for speicifc GPU type, 
# we should remove the label from the old GPU driver version, and add the label to the new GPU driver version for that GPU type.
# If a new GPU driver version is added, we will need to create an entry for it for each GPU type which supports this GPU driver.
# If a new GPU type is added, we will need to create a entry under the gpu_driver_version_info and added the driver/label related information.

gpu_driver_version_info: [
    {
        gpu_device: {
            gpu_type: "NVIDIA_TESLA_V100",
        },
        supported_driver_versions: [
            {
                label: "DEFAULT",
                version: "535.183.01"
            },
            {
                label: "LATEST",
                version: "550.90.07"
            },
            {
                label: "R550",
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_TESLA_A100",
        },
        supported_driver_versions: [
            {
                label: "DEFAULT",
                version: "535.183.01"
            },
            {
                label: "LATEST",
                version: "550.90.07"
            },
            {
                label: "R550",
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_A100_80GB",
        },
        supported_driver_versions: [
            {
                label: "DEFAULT",
                version: "535.183.01"
            },
            {
                label: "LATEST",
                version: "550.90.07"
            },
            {
                label: "R550",
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_H100_80GB",
        },
        supported_driver_versions: [
            {
                label: "DEFAULT",
                version: "535.183.01"
            },
            {
                label: "LATEST",
                version: "550.90.07"
            },
            {
                label: "R550",
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_TESLA_P100",
        },
        supported_driver_versions: [
            {
                label: "DEFAULT",
                version: "535.183.01"
            },
            {
                label: "LATEST",
                version: "550.90.07"
            },
            {
                label: "R550",
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_L4",
        },
        supported_driver_versions: [
            {
                label: "DEFAULT",
                version: "535.183.01"
            },
            {
                label: "LATEST",
                version: "550.90.07"
            },
            {
                label: "R550",
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_TESLA_P4",
        },
        supported_driver_versions: [
            {
                label: "DEFAULT",
                version: "535.183.01"
            },
            {
                label: "LATEST",
                version: "550.90.07"
            },
            {
                label: "R550",
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NVIDIA_TESLA_T4",
        },
        supported_driver_versions: [
            {
                label: "DEFAULT",
                version: "535.183.01"
            },
            {
                label: "LATEST",
                version: "550.90.07"
            },
            {
                label: "R550",
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "OTHERS",
        },
        supported_driver_versions: [
            {
                label: "DEFAULT",
                version: "535.183.01"
            },
            {
                label: "LATEST",
                version: "550.90.07"
            },
            {
                label: "R550",
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            }
        ]
    },
    {
        gpu_device: {
            gpu_type: "NO_GPU",
        },
        supported_driver_versions: [
            {
                label: "DEFAULT",
                version: "535.183.01"
            },
            {
                label: "LATEST",
                version: "550.90.07"
            },
            {
                label: "R550",
                version: "550.90.07"
            },
            {
                version: "550.54.15"
            },
            {
                version: "550.54.14"
            },
            {
                label: "R535",
                version: "535.183.01"
            },
            {
                version: "535.161.08"
            },
            {
                version: "535.161.07"
            },
            {
                version: "535.154.05"
            },
            {
                version: "535.129.03"
            },
            {
                version: "535.104.12"
            },
            {
                version: "535.104.05"
            }
        ]
    }
]