mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-08-12 13:31:52 +00:00
scheduler_perf: more realistic structured parameters tests
Real devices are likely to have a handful of attributes and (for GPUs) the memory as capacity. Most keys will be driver specific, a few may eventually have a domain (none standardized right now).
This commit is contained in:
parent
b071443187
commit
7bbb3465e5
@ -23,11 +23,13 @@ import (
|
|||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
resourceapi "k8s.io/api/resource/v1alpha3"
|
resourceapi "k8s.io/api/resource/v1alpha3"
|
||||||
|
"k8s.io/apimachinery/pkg/api/resource"
|
||||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||||
"k8s.io/client-go/util/workqueue"
|
"k8s.io/client-go/util/workqueue"
|
||||||
"k8s.io/klog/v2"
|
"k8s.io/klog/v2"
|
||||||
draapp "k8s.io/kubernetes/test/e2e/dra/test-driver/app"
|
draapp "k8s.io/kubernetes/test/e2e/dra/test-driver/app"
|
||||||
"k8s.io/kubernetes/test/utils/ktesting"
|
"k8s.io/kubernetes/test/utils/ktesting"
|
||||||
|
"k8s.io/utils/ptr"
|
||||||
)
|
)
|
||||||
|
|
||||||
// createResourceClaimsOp defines an op where resource claims are created.
|
// createResourceClaimsOp defines an op where resource claims are created.
|
||||||
@ -248,7 +250,17 @@ func resourceSlice(driverName, nodeName string, capacity int) *resourceapi.Resou
|
|||||||
slice.Spec.Devices = append(slice.Spec.Devices,
|
slice.Spec.Devices = append(slice.Spec.Devices,
|
||||||
resourceapi.Device{
|
resourceapi.Device{
|
||||||
Name: fmt.Sprintf("instance-%d", i),
|
Name: fmt.Sprintf("instance-%d", i),
|
||||||
Basic: &resourceapi.BasicDevice{},
|
Basic: &resourceapi.BasicDevice{
|
||||||
|
Attributes: map[resourceapi.QualifiedName]resourceapi.DeviceAttribute{
|
||||||
|
"model": {StringValue: ptr.To("A100")},
|
||||||
|
"family": {StringValue: ptr.To("GPU")},
|
||||||
|
"driverVersion": {VersionValue: ptr.To("1.2.3")},
|
||||||
|
"dra.example.com/numa": {IntValue: ptr.To(int64(i))},
|
||||||
|
},
|
||||||
|
Capacity: map[resourceapi.QualifiedName]resource.Quantity{
|
||||||
|
"memory": resource.MustParse("1Gi"),
|
||||||
|
},
|
||||||
|
},
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user