Skip to content

Commit df0a612

Browse files
authored
fix: Update WorkerCudaUpLimitTflopsEnv to use limits.Tflops.Value() instead of info.Fp16TFlops.Value() (#98)
1 parent fdf79ff commit df0a612

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

internal/worker/worker.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ func (wg *WorkerGenerator) GenerateWorkerPod(
102102
Value: strconv.Itoa(port),
103103
}, corev1.EnvVar{
104104
Name: constants.WorkerCudaUpLimitTflopsEnv,
105-
Value: strconv.FormatInt(info.Fp16TFlops.Value(), 10),
105+
Value: strconv.FormatInt(limits.Tflops.Value(), 10),
106106
}, corev1.EnvVar{
107107
Name: constants.WorkerCudaUpLimitEnv,
108108
Value: strconv.FormatInt(int64(math.Ceil(float64(limits.Tflops.Value())/float64(info.Fp16TFlops.Value())*100)), 10),

0 commit comments

Comments
 (0)