fix: default GPU resource to 0 if not present (#1789)

Signed-off-by: Artur Troian <[email protected]>
akash-network · Mar 23, 2023 · 86a9747 · 86a9747
1 parent a7e3cd1
commit 86a9747
Showing 1 changed file with 8 additions and 17 deletions.
diff --git a/sdl/resources.go b/sdl/resources.go
@@ -11,23 +11,6 @@ type v2ComputeResources struct {
 	Storage v2ResourceStorageArray `yaml:"storage"`
 }
 
-// func (sdl *v2ComputeResources) UnmarshalYAML(node *yaml.Node) error {
-// 	res := v2ComputeResources{}
-//
-// 	if err := node.Decode(&res); err != nil {
-// 		return err
-// 	}
-//
-// 	if res.GPU == nil {
-// 		res.GPU = &v2ResourceGPU {
-// 			Units: 0,
-// 		}
-// 	}
-//
-// 	*sdl = res
-// 	return nil
-// }
-
 func (sdl *v2ComputeResources) toDGroupResourceUnits() types.ResourceUnits {
 	if sdl == nil {
 		return types.ResourceUnits{}
@@ -47,6 +30,10 @@ func (sdl *v2ComputeResources) toDGroupResourceUnits() types.ResourceUnits {
 			Units:      types.NewResourceValue(uint64(sdl.GPU.Units)),
 			Attributes: types.Attributes(sdl.GPU.Attributes),
 		}
+	} else {
+		units.GPU = &types.GPU{
+			Units: types.NewResourceValue(0),
+		}
 	}
 
 	if sdl.Memory != nil {
@@ -82,6 +69,10 @@ func toManifestResources(res *v2ComputeResources) types.ResourceUnits {
 		units.GPU = &types.GPU{
 			Units: types.NewResourceValue(uint64(res.GPU.Units)),
 		}
+	} else {
+		units.GPU = &types.GPU{
+			Units: types.NewResourceValue(0),
+		}
 	}
 
 	if res.Memory != nil {