fix: disable ngpu mode by default (#398)

* fix: disable ngpu mode by default

* chore: lint

Author: Code2Life

internal/constants/env.go

@@ -16,6 +16,8 @@ const (
 
 	NvidiaOperatorProgressiveMigrationEnv = "NVIDIA_OPERATOR_PROGRESSIVE_MIGRATION"
 	RunHypervisorUtilGPUAllocatable       = "RUN_HYPERVISOR_UTIL_GPU_ALLOCATABLE"
+
+	UsingCommercialComponentEnv = "COMMERCIAL_PLAN"
 )
 
 // General envs used in compose components manifest

internal/utils/compose.go

@@ -338,11 +338,15 @@ func AddTFDefaultClientConfBeforePatch(
 			}, v1.EnvVar{
 				Name:  constants.HypervisorPortEnv,
 				Value: strconv.Itoa(int(getHypervisorPortNumber(pool.Spec.ComponentConfig.Hypervisor))),
-			}, v1.EnvVar{
-				Name:  constants.NGPUPathEnv,
-				Value: constants.NGPUPathValue,
 			})
 
+			if IsLicensed() {
+				envList = append(envList, v1.EnvVar{
+					Name:  constants.NGPUPathEnv,
+					Value: constants.NGPUPathValue,
+				})
+			}
+
 			// disable GPU limiter killer switch
 			if pod.Annotations[constants.DisableFeaturesAnnotation] != "" {
 				envList = convertDisabledFeaturesToEnvs(pod.Annotations[constants.DisableFeaturesAnnotation], envList)

internal/utils/config.go

@@ -166,7 +166,19 @@ func GetSelfServiceAccountNameShort() string {
 	return parts[len(parts)-1]
 }
 
-var nvidiaOperatorProgressiveMigrationEnv = os.Getenv(constants.NvidiaOperatorProgressiveMigrationEnv) == "true"
+var nvidiaOperatorProgressiveMigrationEnv = os.Getenv(constants.NvidiaOperatorProgressiveMigrationEnv) == constants.TrueStringValue
+
+var isLicensedEnv = os.Getenv(constants.UsingCommercialComponentEnv) == constants.TrueStringValue
+
+func init() {
+	if isLicensedEnv {
+		ctrl.Log.Info("Enabling none open source components, please make sure you are in trial stage or have bought commercial license. Contact us: [email protected]")
+	}
+}
+
+func IsLicensed() bool {
+	return isLicensedEnv
+}
 
 func IsProgressiveMigration() bool {
 	return nvidiaOperatorProgressiveMigrationEnv