Handle MPS node initialization

nebuly-ai · Apr 4, 2023 · b756568 · b756568
1 parent ccc7511
commit b756568
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/controllers/gpupartitioner/node_controller.go b/internal/controllers/gpupartitioner/node_controller.go
@@ -93,6 +93,10 @@ func (c *NodeController) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.
 			return ctrl.Result{}, fmt.Errorf("failed to initialize node MIG partitioning: %w", err)
 		}
 	}
+	// Handle MPS node initialization
+	if gpu.IsMpsPartitioningEnabled(instance) && !nodeInitialized {
+		nodeInitialized = true
+	}
 
 	// If the node is not initialized, do not add it to cluster state
 	if !nodeInitialized {