TAS: allow to dump the usage of domains for TAS ResourceFlavors

pkg/cache/snapshot.go

@@ -79,6 +79,19 @@ func (s *Snapshot) Log(log logr.Logger) {
 			"usage", cohort.ResourceNode.Usage,
 		)
 	}
+
+	// Dump TAS snapshots if the feature is enabled
+	if features.Enabled(features.TopologyAwareScheduling) {
+		for cqName, cq := range s.ClusterQueues {
+			for tasFlavor, tasSnapshot := range cq.TASFlavors {
+				log.Info("TAS Snapshot Free Capacity",
+					"clusterQueue", cqName,
+					"resourceFlavor", tasFlavor,
+					"freeCapacityPerDomain", tasSnapshot.FreeCapacityPerDomain(),
+				)
+			}
+		}
+	}
 }
 
 func (c *Cache) Snapshot(ctx context.Context) (*Snapshot, error) {

pkg/cache/tas_flavor_snapshot.go

@@ -241,6 +241,19 @@ func (s *TASFlavorSnapshot) removeTASUsage(domainID utiltas.TopologyDomainID, us
 	s.leaves[domainID].tasUsage.Sub(usage)
 }
 
+func (s *TASFlavorSnapshot) FreeCapacityPerDomain() map[string]resources.Requests {
+	freeCapacityPerDomain := make(map[string]resources.Requests)
+
+	for domainID, leaf := range s.leaves {
+		// CPU is represented in millicores
+		// Memory is represented in bytes
+		// For other resources, use their raw integer value
+		freeCapacityPerDomain[string(domainID)] = leaf.freeCapacity.Clone()
+	}
+
+	return freeCapacityPerDomain
+}
+
 type TASPodSetRequests struct {
 	PodSet            *kueue.PodSet
 	SinglePodRequests resources.Requests

pkg/cache/tas_flavor_snapshot_test.go

@@ -0,0 +1,63 @@
+/*
+Copyright The Kubernetes Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package cache
+
+import (
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+	corev1 "k8s.io/api/core/v1"
+
+	"sigs.k8s.io/kueue/pkg/resources"
+)
+
+func TestFreeCapacityPerDomain(t *testing.T) {
+	snapshot := &TASFlavorSnapshot{
+		leaves: leafDomainByID{
+			"domain1": &leafDomain{
+				freeCapacity: resources.Requests{
+					corev1.ResourceCPU:    2000,
+					corev1.ResourceMemory: 4 * 1024 * 1024 * 1024, // 4 GiB
+					"nvidia.com/gpu":      1,
+				},
+			},
+			"domain2": &leafDomain{
+				freeCapacity: resources.Requests{
+					corev1.ResourceCPU:    1000,
+					corev1.ResourceMemory: 2 * 1024 * 1024 * 1024, // 2 GiB
+				},
+			},
+		},
+	}
+
+	expected := map[string]resources.Requests{
+		"domain1": {
+			corev1.ResourceCPU:    2000,
+			corev1.ResourceMemory: 4 * 1024 * 1024 * 1024,
+			"nvidia.com/gpu":      1,
+		},
+		"domain2": {
+			corev1.ResourceCPU:    1000,
+			corev1.ResourceMemory: 2 * 1024 * 1024 * 1024,
+		},
+	}
+
+	actual := snapshot.FreeCapacityPerDomain()
+	if diff := cmp.Diff(expected, actual); diff != "" {
+		t.Errorf("FreeCapacityPerDomain() mismatch (-expected +actual):\n%s", diff)
+	}
+}