Adding podaffinity and resourcerequirments to IG Spec (kserve#2711)

* Adding podaffinity and resourcerequirments to IG Spec Signed-off-by: rachitchauhan43 <[email protected]> * Marking Resources and Affinity as optional Signed-off-by: rachitchauhan43 <[email protected]> * Adding first test for ig controller. Signed-off-by: rachitchauhan43 <[email protected]> * Adding more tests for ig controller. Signed-off-by: rachitchauhan43 <[email protected]> * Adding more tests for ig controller. Signed-off-by: rachitchauhan43 <[email protected]> * Adding more tests for ig controller. Signed-off-by: rachitchauhan43 <[email protected]> * Addressing review comments Signed-off-by: rachitchauhan43 <[email protected]> * Removing commented code Signed-off-by: rachitchauhan43 <[email protected]> * addressing review comments Signed-off-by: rachitchauhan43 <[email protected]> --------- Signed-off-by: rachitchauhan43 <[email protected]>
zillow · Mar 4, 2023 · aed4d5b · aed4d5b
1 parent c24ab6d
commit aed4d5b
Show file tree

Hide file tree

Showing 8 changed files with 1,369 additions and 11 deletions.
diff --git a/config/crd/serving.kserve.io_inferencegraphs.yaml b/config/crd/serving.kserve.io_inferencegraphs.yaml
diff --git a/pkg/apis/serving/v1alpha1/inference_graph.go b/pkg/apis/serving/v1alpha1/inference_graph.go
@@ -17,6 +17,7 @@ limitations under the License.
 package v1alpha1
 
 import (
+	corev1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"knative.dev/pkg/apis"
 	duckv1 "knative.dev/pkg/apis/duck/v1"
@@ -45,6 +46,10 @@ type InferenceGraphSpec struct {
 	// Map of InferenceGraph router nodes
 	// Each node defines the router which can be different routing types
 	Nodes map[string]InferenceRouter `json:"nodes"`
+	// +optional
+	Resources corev1.ResourceRequirements `json:"resources,omitempty"`
+	// +optional
+	Affinity *corev1.Affinity `json:"affinity,omitempty" protobuf:"bytes,18,opt,name=affinity"`
 }
 
 // InferenceRouterType constant for inference routing types

diff --git a/pkg/apis/serving/v1alpha1/zz_generated.deepcopy.go b/pkg/apis/serving/v1alpha1/zz_generated.deepcopy.go
diff --git a/pkg/constants/constants.go b/pkg/constants/constants.go
@@ -50,6 +50,7 @@ var (
 // InferenceGraph Constants
 const (
 	RouterHeadersPropagateEnvVar = "PROPAGATE_HEADERS"
+	InferenceGraphLabel          = "serving.kserve.io/inferencegraph"
 )
 
 // TrainedModel Constants