From b756568e52be993e8ec05b7b58b2c8194433a3f4 Mon Sep 17 00:00:00 2001 From: Michele Zanotti Date: Tue, 4 Apr 2023 13:39:42 +0200 Subject: [PATCH] Handle MPS node initialization --- internal/controllers/gpupartitioner/node_controller.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/internal/controllers/gpupartitioner/node_controller.go b/internal/controllers/gpupartitioner/node_controller.go index f7486d4..8a9b9e5 100644 --- 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 {