Fixed enable SGX on exposed services

grycap · Oct 31, 2023 · 9933718 · 9933718
1 parent 83222e9
commit 9933718
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/pkg/backends/k8s.go b/pkg/backends/k8s.go
@@ -127,6 +127,7 @@ func (k *KubeBackend) CreateService(service types.Service) error {
 			MaxScale:     service.Expose.MaxScale,
 			MinScale:     service.Expose.MinScale,
 			CpuThreshold: service.Expose.CpuThreshold,
+			EnableSGX:    service.EnableSGX,
 		}
 		utils.CreateExpose(exposeConf, k.kubeClientset, *k.config)
 	}

diff --git a/pkg/utils/expose.go b/pkg/utils/expose.go
@@ -42,7 +42,6 @@ type Expose struct {
 	Port         int   ` binding:"required" default:"80"`
 	CpuThreshold int32 `default:"80"`
 	EnableSGX    bool
-	EnableGPU    bool
 }
 
 // / Main function that creates all the kubernetes components
@@ -232,6 +231,8 @@ func getPodTemplateSpec(e Expose) v1.PodTemplateSpec {
 						Requests: v1.ResourceList{
 							"cpu": *cores,
 						},
+						// Empty Limits list initialized in case enabling SGX is needed
+						Limits: v1.ResourceList{},
 					},
 				},
 			},
@@ -244,11 +245,6 @@ func getPodTemplateSpec(e Expose) v1.PodTemplateSpec {
 		template.Spec.Containers[0].Resources.Limits["sgx.intel.com/enclave"] = sgx
 	}
 
-	if e.EnableGPU {
-		gpu, _ := resource.ParseQuantity("1")
-		template.Spec.Containers[0].Resources.Limits["nvidia.com/gpu"] = gpu
-	}
-
 	return template
 }