Skip to content

Commit

Permalink
remove cycle
Browse files Browse the repository at this point in the history
  • Loading branch information
ishaansehgal99 committed Apr 4, 2024
1 parent 53394ee commit c3ba6ae
Showing 1 changed file with 1 addition and 7 deletions.
8 changes: 1 addition & 7 deletions pkg/utils/common-preset.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,7 @@ package utils
import (
"fmt"
kaitov1alpha1 "github.com/azure/kaito/api/v1alpha1"
"github.com/azure/kaito/pkg/tuning"
corev1 "k8s.io/api/core/v1"
"strconv"
)

const (
Expand Down Expand Up @@ -70,11 +68,7 @@ func GetInstanceGPUCount(wObj *kaitov1alpha1.Workspace) int {
sku := wObj.Resource.InstanceType
gpuConfig, exists := kaitov1alpha1.SupportedGPUConfigs[sku]
if !exists {
numProcesses, err := strconv.Atoi(tuning.DefaultNumProcesses)
if err != nil {
return 1
}
return numProcesses
return 1
}
return gpuConfig.GPUCount
}
Expand Down

0 comments on commit c3ba6ae

Please sign in to comment.