VRAMAvailablePercentage added (#10)

* vram available added * type cange to int64 * Dup fix
spheronFdn · Nov 11, 2024 · bd4d859 · bd4d859
1 parent 202908e
commit bd4d859
Show file tree

Hide file tree

Showing 4 changed files with 97 additions and 50 deletions.
diff --git a/docs/proto/provider.md b/docs/proto/provider.md
@@ -965,6 +965,7 @@
  | `modelid` | [string](#string) |  |  |
  | `interface` | [string](#string) |  |  |
  | `memory_size` | [string](#string) |  |  |
+ | `vram_available_percentage` | [int64](#int64) |  |  |
 
 
 

diff --git a/go/inventory/v1/gpu.go b/go/inventory/v1/gpu.go
@@ -25,12 +25,13 @@ func (r *GPU) Dup() GPU {
 
 func (s *GPUInfo) Dup() GPUInfo {
 	res := GPUInfo{
-		Vendor:     s.Vendor,
-		VendorID:   s.VendorID,
-		Name:       s.Name,
-		ModelID:    s.ModelID,
-		Interface:  s.Interface,
-		MemorySize: s.MemorySize,
+		Vendor:               s.Vendor,
+		VendorID:             s.VendorID,
+		Name:                 s.Name,
+		ModelID:              s.ModelID,
+		Interface:            s.Interface,
+		MemorySize:           s.MemorySize,
+		VRAMAvailablePercentage: s.VRAMAvailablePercentage,
 	}
 
 	return res
@@ -45,11 +46,12 @@ func (s GPUInfoS) Dup() GPUInfoS {
 
 	for _, n := range s {
 		res = append(res, GPUInfo{
-			Vendor:     n.Vendor,
-			Name:       n.Name,
-			ModelID:    n.ModelID,
-			Interface:  n.Interface,
-			MemorySize: n.MemorySize,
+			Vendor:               n.Vendor,
+			Name:                 n.Name,
+			ModelID:              n.ModelID,
+			Interface:            n.Interface,
+			MemorySize:           n.MemorySize,
+			VRAMAvailablePercentage: n.VRAMAvailablePercentage,
 		})
 	}
 

diff --git a/go/inventory/v1/gpu.pb.go b/go/inventory/v1/gpu.pb.go
diff --git a/proto/provider/akash/inventory/v1/gpu.proto b/proto/provider/akash/inventory/v1/gpu.proto
@@ -41,6 +41,11 @@ message GPUInfo {
     (gogoproto.jsontag)      = "memory_size",
     (gogoproto.moretags)     = "yaml:\"memory_size\""
   ];
+  int64 vram_available_percentage = 7 [
+    (gogoproto.customname)   = "VRAMAvailablePercentage",
+    (gogoproto.jsontag)      = "vram_available_percentage",
+    (gogoproto.moretags)     = "yaml:\"vram_available_percentage\""
+  ];
 }
 
 // GPUInfo reports GPU inventory details