cos/tools: add proto schema for gpu_driver_versions
BUG=b/328517894
TEST=test locally, cloudbuild test
Change-Id: Ia049f4d2139bfaacbb4fe42d70efa29dab454ac4
Reviewed-on: https://cos-review.googlesource.com/c/cos/tools/+/66474
Cloud-Build: GCB Service account <228075978874@cloudbuild.gserviceaccount.com>
Reviewed-by: Arnav Kansal <rnv@google.com>
Reviewed-by: Oleksandr Tymoshenko <ovt@google.com>
Reviewed-by: Robert Kolchmeyer <rkolchmeyer@google.com>
Tested-by: Shuo Yang <gshuoy@google.com>
diff --git a/src/pkg/gpuconfig/config.go b/src/pkg/gpuconfig/config.go
index b188c43..a5e7a55 100644
--- a/src/pkg/gpuconfig/config.go
+++ b/src/pkg/gpuconfig/config.go
@@ -5,7 +5,7 @@
// the COS Image and the COS Kernel CI.
package gpuconfig
-//go:generate protoc --go_out=:./pb -I. proto/config.proto
+//go:generate protoc --go_out=:./pb -I. proto/config.proto proto/gpu_driver_versions.proto
import (
"time"
diff --git a/src/pkg/gpuconfig/pb/config.pb.go b/src/pkg/gpuconfig/pb/config.pb.go
index 620b4a5..ef66c61 100644
--- a/src/pkg/gpuconfig/pb/config.pb.go
+++ b/src/pkg/gpuconfig/pb/config.pb.go
@@ -1,7 +1,7 @@
// Code generated by protoc-gen-go. DO NOT EDIT.
// versions:
-// protoc-gen-go v1.28.0
-// protoc v3.17.3
+// protoc-gen-go v1.31.0
+// protoc v3.21.12
// source: proto/config.proto
package pb
diff --git a/src/pkg/gpuconfig/pb/gpu_driver_versions.pb.go b/src/pkg/gpuconfig/pb/gpu_driver_versions.pb.go
new file mode 100644
index 0000000..842d60b
--- /dev/null
+++ b/src/pkg/gpuconfig/pb/gpu_driver_versions.pb.go
@@ -0,0 +1,308 @@
+// gpu_driver_versions.proto schema
+
+// Code generated by protoc-gen-go. DO NOT EDIT.
+// versions:
+// protoc-gen-go v1.31.0
+// protoc v3.21.12
+// source: proto/gpu_driver_versions.proto
+
+package pb
+
+import (
+ protoreflect "google.golang.org/protobuf/reflect/protoreflect"
+ protoimpl "google.golang.org/protobuf/runtime/protoimpl"
+ reflect "reflect"
+ sync "sync"
+)
+
+const (
+ // Verify that this generated code is sufficiently up-to-date.
+ _ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
+ // Verify that runtime/protoimpl is sufficiently up-to-date.
+ _ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
+)
+
+// Represents all the information about a GPU driver version.
+// It contains the GPU driver version label (e.g., "default", "latest"), and the GPU driver version that is associated with that label.
+type DriverVersion struct {
+ state protoimpl.MessageState
+ sizeCache protoimpl.SizeCache
+ unknownFields protoimpl.UnknownFields
+
+ // GPU driver version label.
+ Label string `protobuf:"bytes,1,opt,name=label,proto3" json:"label,omitempty"`
+ // The driver version that is associated with this label.
+ Version string `protobuf:"bytes,2,opt,name=version,proto3" json:"version,omitempty"` // more metadata about the driver version.
+}
+
+func (x *DriverVersion) Reset() {
+ *x = DriverVersion{}
+ if protoimpl.UnsafeEnabled {
+ mi := &file_proto_gpu_driver_versions_proto_msgTypes[0]
+ ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+ ms.StoreMessageInfo(mi)
+ }
+}
+
+func (x *DriverVersion) String() string {
+ return protoimpl.X.MessageStringOf(x)
+}
+
+func (*DriverVersion) ProtoMessage() {}
+
+func (x *DriverVersion) ProtoReflect() protoreflect.Message {
+ mi := &file_proto_gpu_driver_versions_proto_msgTypes[0]
+ if protoimpl.UnsafeEnabled && x != nil {
+ ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+ if ms.LoadMessageInfo() == nil {
+ ms.StoreMessageInfo(mi)
+ }
+ return ms
+ }
+ return mi.MessageOf(x)
+}
+
+// Deprecated: Use DriverVersion.ProtoReflect.Descriptor instead.
+func (*DriverVersion) Descriptor() ([]byte, []int) {
+ return file_proto_gpu_driver_versions_proto_rawDescGZIP(), []int{0}
+}
+
+func (x *DriverVersion) GetLabel() string {
+ if x != nil {
+ return x.Label
+ }
+ return ""
+}
+
+func (x *DriverVersion) GetVersion() string {
+ if x != nil {
+ return x.Version
+ }
+ return ""
+}
+
+// Represents GPU driver version information for a specific GPU type, mapping from GPU type to GPU driver version information.
+type GPUDriverVersionInfo struct {
+ state protoimpl.MessageState
+ sizeCache protoimpl.SizeCache
+ unknownFields protoimpl.UnknownFields
+
+ // GPU type.
+ GpuType string `protobuf:"bytes,1,opt,name=gpu_type,json=gpuType,proto3" json:"gpu_type,omitempty"`
+ // A repeated field to hold list of GPU driver versions with the associated labels that are built by cos.
+ // it is something like: <{"latest","535.129.11"}, ...>
+ SupportedDriverVersions []*DriverVersion `protobuf:"bytes,2,rep,name=supported_driver_versions,json=supportedDriverVersions,proto3" json:"supported_driver_versions,omitempty"`
+}
+
+func (x *GPUDriverVersionInfo) Reset() {
+ *x = GPUDriverVersionInfo{}
+ if protoimpl.UnsafeEnabled {
+ mi := &file_proto_gpu_driver_versions_proto_msgTypes[1]
+ ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+ ms.StoreMessageInfo(mi)
+ }
+}
+
+func (x *GPUDriverVersionInfo) String() string {
+ return protoimpl.X.MessageStringOf(x)
+}
+
+func (*GPUDriverVersionInfo) ProtoMessage() {}
+
+func (x *GPUDriverVersionInfo) ProtoReflect() protoreflect.Message {
+ mi := &file_proto_gpu_driver_versions_proto_msgTypes[1]
+ if protoimpl.UnsafeEnabled && x != nil {
+ ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+ if ms.LoadMessageInfo() == nil {
+ ms.StoreMessageInfo(mi)
+ }
+ return ms
+ }
+ return mi.MessageOf(x)
+}
+
+// Deprecated: Use GPUDriverVersionInfo.ProtoReflect.Descriptor instead.
+func (*GPUDriverVersionInfo) Descriptor() ([]byte, []int) {
+ return file_proto_gpu_driver_versions_proto_rawDescGZIP(), []int{1}
+}
+
+func (x *GPUDriverVersionInfo) GetGpuType() string {
+ if x != nil {
+ return x.GpuType
+ }
+ return ""
+}
+
+func (x *GPUDriverVersionInfo) GetSupportedDriverVersions() []*DriverVersion {
+ if x != nil {
+ return x.SupportedDriverVersions
+ }
+ return nil
+}
+
+// Represents a list of GPU driver version information.
+type GPUDriverVerionInfoList struct {
+ state protoimpl.MessageState
+ sizeCache protoimpl.SizeCache
+ unknownFields protoimpl.UnknownFields
+
+ GpuDriverVersionInfo []*GPUDriverVersionInfo `protobuf:"bytes,1,rep,name=gpu_driver_version_info,json=gpuDriverVersionInfo,proto3" json:"gpu_driver_version_info,omitempty"`
+}
+
+func (x *GPUDriverVerionInfoList) Reset() {
+ *x = GPUDriverVerionInfoList{}
+ if protoimpl.UnsafeEnabled {
+ mi := &file_proto_gpu_driver_versions_proto_msgTypes[2]
+ ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+ ms.StoreMessageInfo(mi)
+ }
+}
+
+func (x *GPUDriverVerionInfoList) String() string {
+ return protoimpl.X.MessageStringOf(x)
+}
+
+func (*GPUDriverVerionInfoList) ProtoMessage() {}
+
+func (x *GPUDriverVerionInfoList) ProtoReflect() protoreflect.Message {
+ mi := &file_proto_gpu_driver_versions_proto_msgTypes[2]
+ if protoimpl.UnsafeEnabled && x != nil {
+ ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+ if ms.LoadMessageInfo() == nil {
+ ms.StoreMessageInfo(mi)
+ }
+ return ms
+ }
+ return mi.MessageOf(x)
+}
+
+// Deprecated: Use GPUDriverVerionInfoList.ProtoReflect.Descriptor instead.
+func (*GPUDriverVerionInfoList) Descriptor() ([]byte, []int) {
+ return file_proto_gpu_driver_versions_proto_rawDescGZIP(), []int{2}
+}
+
+func (x *GPUDriverVerionInfoList) GetGpuDriverVersionInfo() []*GPUDriverVersionInfo {
+ if x != nil {
+ return x.GpuDriverVersionInfo
+ }
+ return nil
+}
+
+var File_proto_gpu_driver_versions_proto protoreflect.FileDescriptor
+
+var file_proto_gpu_driver_versions_proto_rawDesc = []byte{
+ 0x0a, 0x1f, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2f, 0x67, 0x70, 0x75, 0x5f, 0x64, 0x72, 0x69, 0x76,
+ 0x65, 0x72, 0x5f, 0x76, 0x65, 0x72, 0x73, 0x69, 0x6f, 0x6e, 0x73, 0x2e, 0x70, 0x72, 0x6f, 0x74,
+ 0x6f, 0x12, 0x02, 0x70, 0x62, 0x22, 0x3f, 0x0a, 0x0d, 0x44, 0x72, 0x69, 0x76, 0x65, 0x72, 0x56,
+ 0x65, 0x72, 0x73, 0x69, 0x6f, 0x6e, 0x12, 0x14, 0x0a, 0x05, 0x6c, 0x61, 0x62, 0x65, 0x6c, 0x18,
+ 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x05, 0x6c, 0x61, 0x62, 0x65, 0x6c, 0x12, 0x18, 0x0a, 0x07,
+ 0x76, 0x65, 0x72, 0x73, 0x69, 0x6f, 0x6e, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x07, 0x76,
+ 0x65, 0x72, 0x73, 0x69, 0x6f, 0x6e, 0x22, 0x80, 0x01, 0x0a, 0x14, 0x47, 0x50, 0x55, 0x44, 0x72,
+ 0x69, 0x76, 0x65, 0x72, 0x56, 0x65, 0x72, 0x73, 0x69, 0x6f, 0x6e, 0x49, 0x6e, 0x66, 0x6f, 0x12,
+ 0x19, 0x0a, 0x08, 0x67, 0x70, 0x75, 0x5f, 0x74, 0x79, 0x70, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28,
+ 0x09, 0x52, 0x07, 0x67, 0x70, 0x75, 0x54, 0x79, 0x70, 0x65, 0x12, 0x4d, 0x0a, 0x19, 0x73, 0x75,
+ 0x70, 0x70, 0x6f, 0x72, 0x74, 0x65, 0x64, 0x5f, 0x64, 0x72, 0x69, 0x76, 0x65, 0x72, 0x5f, 0x76,
+ 0x65, 0x72, 0x73, 0x69, 0x6f, 0x6e, 0x73, 0x18, 0x02, 0x20, 0x03, 0x28, 0x0b, 0x32, 0x11, 0x2e,
+ 0x70, 0x62, 0x2e, 0x44, 0x72, 0x69, 0x76, 0x65, 0x72, 0x56, 0x65, 0x72, 0x73, 0x69, 0x6f, 0x6e,
+ 0x52, 0x17, 0x73, 0x75, 0x70, 0x70, 0x6f, 0x72, 0x74, 0x65, 0x64, 0x44, 0x72, 0x69, 0x76, 0x65,
+ 0x72, 0x56, 0x65, 0x72, 0x73, 0x69, 0x6f, 0x6e, 0x73, 0x22, 0x6a, 0x0a, 0x17, 0x47, 0x50, 0x55,
+ 0x44, 0x72, 0x69, 0x76, 0x65, 0x72, 0x56, 0x65, 0x72, 0x69, 0x6f, 0x6e, 0x49, 0x6e, 0x66, 0x6f,
+ 0x4c, 0x69, 0x73, 0x74, 0x12, 0x4f, 0x0a, 0x17, 0x67, 0x70, 0x75, 0x5f, 0x64, 0x72, 0x69, 0x76,
+ 0x65, 0x72, 0x5f, 0x76, 0x65, 0x72, 0x73, 0x69, 0x6f, 0x6e, 0x5f, 0x69, 0x6e, 0x66, 0x6f, 0x18,
+ 0x01, 0x20, 0x03, 0x28, 0x0b, 0x32, 0x18, 0x2e, 0x70, 0x62, 0x2e, 0x47, 0x50, 0x55, 0x44, 0x72,
+ 0x69, 0x76, 0x65, 0x72, 0x56, 0x65, 0x72, 0x73, 0x69, 0x6f, 0x6e, 0x49, 0x6e, 0x66, 0x6f, 0x52,
+ 0x14, 0x67, 0x70, 0x75, 0x44, 0x72, 0x69, 0x76, 0x65, 0x72, 0x56, 0x65, 0x72, 0x73, 0x69, 0x6f,
+ 0x6e, 0x49, 0x6e, 0x66, 0x6f, 0x42, 0x06, 0x5a, 0x04, 0x2e, 0x3b, 0x70, 0x62, 0x62, 0x06, 0x70,
+ 0x72, 0x6f, 0x74, 0x6f, 0x33,
+}
+
+var (
+ file_proto_gpu_driver_versions_proto_rawDescOnce sync.Once
+ file_proto_gpu_driver_versions_proto_rawDescData = file_proto_gpu_driver_versions_proto_rawDesc
+)
+
+func file_proto_gpu_driver_versions_proto_rawDescGZIP() []byte {
+ file_proto_gpu_driver_versions_proto_rawDescOnce.Do(func() {
+ file_proto_gpu_driver_versions_proto_rawDescData = protoimpl.X.CompressGZIP(file_proto_gpu_driver_versions_proto_rawDescData)
+ })
+ return file_proto_gpu_driver_versions_proto_rawDescData
+}
+
+var file_proto_gpu_driver_versions_proto_msgTypes = make([]protoimpl.MessageInfo, 3)
+var file_proto_gpu_driver_versions_proto_goTypes = []interface{}{
+ (*DriverVersion)(nil), // 0: pb.DriverVersion
+ (*GPUDriverVersionInfo)(nil), // 1: pb.GPUDriverVersionInfo
+ (*GPUDriverVerionInfoList)(nil), // 2: pb.GPUDriverVerionInfoList
+}
+var file_proto_gpu_driver_versions_proto_depIdxs = []int32{
+ 0, // 0: pb.GPUDriverVersionInfo.supported_driver_versions:type_name -> pb.DriverVersion
+ 1, // 1: pb.GPUDriverVerionInfoList.gpu_driver_version_info:type_name -> pb.GPUDriverVersionInfo
+ 2, // [2:2] is the sub-list for method output_type
+ 2, // [2:2] is the sub-list for method input_type
+ 2, // [2:2] is the sub-list for extension type_name
+ 2, // [2:2] is the sub-list for extension extendee
+ 0, // [0:2] is the sub-list for field type_name
+}
+
+func init() { file_proto_gpu_driver_versions_proto_init() }
+func file_proto_gpu_driver_versions_proto_init() {
+ if File_proto_gpu_driver_versions_proto != nil {
+ return
+ }
+ if !protoimpl.UnsafeEnabled {
+ file_proto_gpu_driver_versions_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+ switch v := v.(*DriverVersion); i {
+ case 0:
+ return &v.state
+ case 1:
+ return &v.sizeCache
+ case 2:
+ return &v.unknownFields
+ default:
+ return nil
+ }
+ }
+ file_proto_gpu_driver_versions_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
+ switch v := v.(*GPUDriverVersionInfo); i {
+ case 0:
+ return &v.state
+ case 1:
+ return &v.sizeCache
+ case 2:
+ return &v.unknownFields
+ default:
+ return nil
+ }
+ }
+ file_proto_gpu_driver_versions_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
+ switch v := v.(*GPUDriverVerionInfoList); i {
+ case 0:
+ return &v.state
+ case 1:
+ return &v.sizeCache
+ case 2:
+ return &v.unknownFields
+ default:
+ return nil
+ }
+ }
+ }
+ type x struct{}
+ out := protoimpl.TypeBuilder{
+ File: protoimpl.DescBuilder{
+ GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
+ RawDescriptor: file_proto_gpu_driver_versions_proto_rawDesc,
+ NumEnums: 0,
+ NumMessages: 3,
+ NumExtensions: 0,
+ NumServices: 0,
+ },
+ GoTypes: file_proto_gpu_driver_versions_proto_goTypes,
+ DependencyIndexes: file_proto_gpu_driver_versions_proto_depIdxs,
+ MessageInfos: file_proto_gpu_driver_versions_proto_msgTypes,
+ }.Build()
+ File_proto_gpu_driver_versions_proto = out.File
+ file_proto_gpu_driver_versions_proto_rawDesc = nil
+ file_proto_gpu_driver_versions_proto_goTypes = nil
+ file_proto_gpu_driver_versions_proto_depIdxs = nil
+}
diff --git a/src/pkg/gpuconfig/proto/gpu_driver_versions.proto b/src/pkg/gpuconfig/proto/gpu_driver_versions.proto
new file mode 100644
index 0000000..60fb9bf
--- /dev/null
+++ b/src/pkg/gpuconfig/proto/gpu_driver_versions.proto
@@ -0,0 +1,34 @@
+// gpu_driver_versions.proto schema
+syntax = "proto3";
+
+package pb;
+
+option go_package = ".;pb";
+
+// Represents all the information about a GPU driver version.
+// It contains the GPU driver version label (e.g., "default", "latest"), and the GPU driver version that is associated with that label.
+message DriverVersion {
+ // GPU driver version label.
+ string label = 1;
+ // The driver version that is associated with this label.
+ string version = 2;
+ // more metadata about the driver version.
+}
+
+
+
+// Represents GPU driver version information for a specific GPU type, mapping from GPU type to GPU driver version information.
+message GPUDriverVersionInfo
+{
+ // GPU type.
+ string gpu_type = 1;
+
+ // A repeated field to hold list of GPU driver versions with the associated labels that are built by cos.
+ // it is something like: <{"latest","535.129.11"}, ...>
+ repeated DriverVersion supported_driver_versions = 2;
+ }
+
+// Represents a list of GPU driver version information.
+message GPUDriverVerionInfoList {
+ repeated GPUDriverVersionInfo gpu_driver_version_info = 1;
+}