cuda and nvidia version added (#9)

spheronFdn · Nov 7, 2024 · 202908e · 202908e
1 parent 58d116a
commit 202908e
Show file tree

Hide file tree

Showing 4 changed files with 154 additions and 29 deletions.
diff --git a/docs/proto/provider.md b/docs/proto/provider.md
@@ -863,6 +863,8 @@
  | `name` | [string](#string) |  |  |
  | `resources` | [NodeResources](#akash.inventory.v1.NodeResources) |  |  |
  | `capabilities` | [NodeCapabilities](#akash.inventory.v1.NodeCapabilities) |  |  |
+ | `cuda_version` | [string](#string) |  |  |
+ | `nvidia_driver_version` | [string](#string) |  |  |
 
 
 

diff --git a/go/inventory/v1/node.go b/go/inventory/v1/node.go
@@ -26,6 +26,8 @@ func (nd *Node) Dup() Node {
 		Name:         nd.Name,
 		Resources:    nd.Resources.Dup(),
 		Capabilities: nd.Capabilities.Dup(),
+		CudaVersion:  nd.CudaVersion,
+		NvidiaDriverVersion: nd.NvidiaDriverVersion,
 	}
 
 	return res

diff --git a/go/inventory/v1/node.pb.go b/go/inventory/v1/node.pb.go
diff --git a/proto/provider/akash/inventory/v1/node.proto b/proto/provider/akash/inventory/v1/node.proto
@@ -36,4 +36,18 @@ message Node {
     (gogoproto.jsontag)      = "capabilities",
     (gogoproto.moretags)     = "yaml:\"capabilities\""
   ];
+
+  string cuda_version = 4 [
+    (gogoproto.nullable)     = true,
+    (gogoproto.customname)   = "CudaVersion",
+    (gogoproto.jsontag)      = "cuda_version",
+    (gogoproto.moretags)     = "yaml:\"cuda_version\""
+  ];
+
+  string nvidia_driver_version = 5 [
+    (gogoproto.nullable)     = true,
+    (gogoproto.customname)   = "NvidiaDriverVersion",
+    (gogoproto.jsontag)      = "nvidia_driver_version",
+    (gogoproto.moretags)     = "yaml:\"nvidia_driver_version\""
+  ];
 }