os arch cudaversion nvidiaversion added

spheronFdn · Oct 25, 2024 · 10b6627 · 10b6627
1 parent daba8c9
commit 10b6627
Show file tree

Hide file tree

Showing 5 changed files with 299 additions and 50 deletions.
diff --git a/docs/proto/provider.md b/docs/proto/provider.md
@@ -826,6 +826,10 @@
  | `volumes_mounted` | [ResourcePair](#akash.inventory.v1.ResourcePair) |  |  |
  | `version` | [string](#string) |  |  |
  | `bandwidth` | [string](#string) |  |  |
+ | `os` | [string](#string) |  |  |
+ | `arch` | [string](#string) |  |  |
+ | `cuda_version` | [string](#string) |  |  |
+ | `nvidia_driver_version` | [string](#string) |  |  |
 
 
 

diff --git a/go/inventory/v1/metrics.go b/go/inventory/v1/metrics.go
@@ -15,11 +15,15 @@ type ResourcesMetric struct {
 }
 
 type NodeMetrics struct {
-	Name        string          `json:"name"`
-	Allocatable ResourcesMetric `json:"allocatable"`
-	Available   ResourcesMetric `json:"available"`
-	Bandwidth   string          `json:"bandwidth,omitempty"`
-	Version     string          `json:"version,omitempty"`
+	Name                string          `json:"name"`
+	Allocatable         ResourcesMetric `json:"allocatable"`
+	Available           ResourcesMetric `json:"available"`
+	Bandwidth           string          `json:"bandwidth,omitempty"`
+	Version             string          `json:"version,omitempty"`
+	Os                  string          `json:"os,omitempty"`
+	Arch                string          `json:"arch,omitempty"`
+	CudaVersion         string          `json:"cuda_version,omitempty"`
+	NvidiaDriverVersion string          `json:"nvidia_driver_version,omitempty"`
 }
 
 type Metrics struct {

diff --git a/go/inventory/v1/resources.go b/go/inventory/v1/resources.go
@@ -10,6 +10,10 @@ func (s *NodeResources) Dup() NodeResources {
 		VolumesMounted:   s.VolumesMounted.Dup(),
 		Version:          s.Version,
 		Bandwidth:        s.Bandwidth,
+		Os:               s.Os,
+		Arch:             s.Arch,
+		CudaVersion:      s.CudaVersion,
+		NvidiaDriverVersion: s.NvidiaDriverVersion,
 	}
 
 	return res

diff --git a/go/inventory/v1/resources.pb.go b/go/inventory/v1/resources.pb.go