Skip to content

Commit 0b3fca0

Browse files
hermes-local-thinking-cap-qwen3.6-27bsimisoft-exo
authored andcommitted
fix: add GPU labels to exo limits command
Co-authored-by: hermes-local-thinking-cap-qwen3.6-27b <hermes-local-thinking-cap-qwen3.6-27b@nousresearch.com>
1 parent d555283 commit 0b3fca0

2 files changed

Lines changed: 40 additions & 47 deletions

File tree

cmd/limits.go

Lines changed: 38 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -70,31 +70,47 @@ Supported output template annotations: %s`,
7070
}
7171

7272
resourceLimitLabels := map[string]string{
73-
limitComputeInstances: "Compute instances",
74-
limitDatabases: "Databases",
75-
limitElasticIPs: "Elastic IP addresses",
76-
limitIAMAPIKeys: "IAM API keys",
77-
limitInstanceSnapshots: "Compute instance snapshots",
78-
limitInstanceTemplates: "Compute instance templates",
79-
limitNLB: "Network Load Balancers",
80-
limitPrivateNetworks: "Private networks",
81-
limitSKSClusters: "SKS clusters",
82-
limitSOSBuckets: "SOS buckets",
83-
limitBlockStorageVolumes: "Block Storage Volumes",
84-
limitBlockStorage: "Block Storage cumulative size (GiB)",
85-
limitBlockStorageMaxSize: "Max size of a Block Storage Volume (GiB)",
86-
}
73+
limitComputeInstances: "Compute instances",
74+
limitDatabases: "Databases",
75+
limitElasticIPs: "Elastic IP addresses",
76+
limitIAMAPIKeys: "IAM API keys",
77+
limitInstanceSnapshots: "Compute instance snapshots",
78+
limitInstanceTemplates: "Compute instance templates",
79+
limitNLB: "Network Load Balancers",
80+
limitPrivateNetworks: "Private networks",
81+
limitSKSClusters: "SKS clusters",
82+
limitSOSBuckets: "SOS buckets",
83+
limitBlockStorageVolumes: "Block Storage Volumes",
84+
limitBlockStorage: "Block Storage cumulative size (GiB)",
85+
limitBlockStorageMaxSize: "Max size of a Block Storage Volume (GiB)",
86+
}
87+
88+
gpuResourceLabels := map[string]string{
89+
gpu2: "GPU - GPU2",
90+
gpu3: "GPU - GPU3",
91+
gpua30: "GPU - A30",
92+
gpu3080ti: "GPU - 3080 Ti",
93+
gpua5000: "GPU - A5000",
94+
gpurtx6000pro: "GPU - RTX 6000 Pro",
95+
}
8796

8897
out := LimitsOutput{}
89-
for _, quota := range quotas.Quotas {
90-
if label, ok := resourceLimitLabels[quota.Resource]; ok {
91-
out = append(out, LimitsItemOutput{
92-
Resource: label,
93-
Used: quota.Usage,
94-
Max: quota.Limit,
95-
})
98+
for _, quota := range quotas.Quotas {
99+
if label, ok := resourceLimitLabels[quota.Resource]; ok {
100+
out = append(out, LimitsItemOutput{
101+
Resource: label,
102+
Used: quota.Usage,
103+
Max: quota.Limit,
104+
})
105+
}
106+
if label, ok := gpuResourceLabels[quota.Resource]; ok {
107+
out = append(out, LimitsItemOutput{
108+
Resource: label,
109+
Used: quota.Usage,
110+
Max: quota.Limit,
111+
})
112+
}
96113
}
97-
}
98114

99115
sort.Slice(out, func(i, j int) bool {
100116
return out[i].Resource < out[j].Resource

cmd/limits_gpu.go

Lines changed: 2 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -41,10 +41,10 @@ type LimitsGpuCmd struct {
4141
}
4242

4343
func (c *LimitsGpuCmd) CmdAliases() []string { return nil }
44-
func (c *LimitsGpuCmd) CmdShort() string { return "Show all limits including per-family GPU limits" }
44+
func (c *LimitsGpuCmd) CmdShort() string { return "Show per-family GPU limits" }
4545
func (c *LimitsGpuCmd) CmdLong() string {
4646
return strings.Join([]string{
47-
"Show all account limits, including per-family GPU quotas (A5000, A30, 3080 Ti, etc.).",
47+
"Show per-family GPU quotas (A5000, A30, 3080 Ti, RTX 6000 Pro, GPU2, GPU3).",
4848
"",
4949
fmt.Sprintf("Supported output template annotations: %s",
5050
strings.Join(output.TemplateAnnotations(&GpuLimitsOutput{}), ", ")),
@@ -68,31 +68,8 @@ func (c *LimitsGpuCmd) CmdRun(_ *cobra.Command, _ []string) error {
6868
return err
6969
}
7070

71-
resourceLimitLabels := map[string]string{
72-
limitComputeInstances: "Compute instances",
73-
limitDatabases: "Databases",
74-
limitElasticIPs: "Elastic IP addresses",
75-
limitIAMAPIKeys: "IAM API keys",
76-
limitInstanceSnapshots: "Compute instance snapshots",
77-
limitInstanceTemplates: "Compute instance templates",
78-
limitNLB: "Network Load Balancers",
79-
limitPrivateNetworks: "Private networks",
80-
limitSKSClusters: "SKS clusters",
81-
limitSOSBuckets: "SOS buckets",
82-
limitBlockStorageVolumes: "Block Storage Volumes",
83-
limitBlockStorage: "Block Storage cumulative size (GiB)",
84-
limitBlockStorageMaxSize: "Max size of a Block Storage Volume (GiB)",
85-
}
86-
8771
out := GpuLimitsOutput{}
8872
for _, quota := range quotas.Quotas {
89-
if label, ok := resourceLimitLabels[quota.Resource]; ok {
90-
out = append(out, GpuLimitOutput{
91-
Resource: label,
92-
Used: quota.Usage,
93-
Max: quota.Limit,
94-
})
95-
}
9673
if label, ok := gpuResourceLabels[quota.Resource]; ok {
9774
out = append(out, GpuLimitOutput{
9875
Resource: label,

0 commit comments

Comments
 (0)