add basic gpu support

Author: zlangbert

internal/commands/deploy.go

@@ -27,6 +27,7 @@ var Deploy = &cli.Command{
 		containerPortFlag,
 		requestedMemoryFlag,
 		requestedCPUFlag,
+		requestedGPUFlag,
 		additionalContainerPortsFlag,
 		envVarsFlag,
 		idleTimeoutFlag,
@@ -77,6 +78,7 @@ var Deploy = &cli.Command{
 				ContainerPort:            deploy.ContainerPort,
 				RequestedMemoryMB:        deploy.RequestedMemoryMB,
 				RequestedCPU:             deploy.RequestedCPU,
+				ExperimentalRequestedGPU: &deploy.RequestedGPU, // TODO: add support for final API field when it exists
 				AdditionalContainerPorts: deploy.AdditionalContainerPorts,
 				Env:                      deploy.Env,
 				DeploymentTag:            deploymentTag,
@@ -149,6 +151,11 @@ func (c *DeployConfig) Merge(latest *components.DeploymentV3, isIdleTimeoutDefau
 		c.RequestedCPU = latest.RequestedCPU
 	}
 
+	if c.RequestedGPU == 0 {
+		// TODO: add support for final API field when it exists
+		c.RequestedGPU = *latest.ExperimentalRequestedGPU
+	}
+
 	if len(c.AdditionalContainerPorts) == 0 {
 		c.AdditionalContainerPorts = latest.AdditionalContainerPorts
 	}

internal/commands/deployment.go

@@ -181,6 +181,7 @@ var Deployment = &cli.Command{
 						ContainerPort:            deployment.ContainerPort,
 						RequestedMemoryMB:        deployment.RequestedMemoryMB,
 						RequestedCPU:             deployment.RequestedCPU,
+						ExperimentalRequestedGPU: &deployment.RequestedGPU, // TODO: add support for final API field when it exists
 						AdditionalContainerPorts: deployment.AdditionalContainerPorts,
 						Env:                      deployment.Env,
 						DeploymentTag:            deploymentTag,
@@ -297,6 +298,17 @@ var (
 		Category:   "Deployment:",
 	}
 
+	requestedGPUFlag = &workaround.FloatFlag{
+		Name: "requested-gpu",
+		Sources: cli.NewValueSourceChain(
+			cli.EnvVar(deploymentEnvVar("REQUESTED_GPU")),
+			altsrc.ConfigFile(configFlag.Name, "deployment.requested-gpu"),
+		),
+		Usage:      "`<gpus>` to allocate to your process",
+		Persistent: true,
+		Category:   "Deployment:",
+	}
+
 	fromLatestFlag = &cli.BoolFlag{
 		Name:     "from-latest",
 		Sources:  cli.EnvVars(deploymentEnvVar("FROM_LATEST")),
@@ -457,6 +469,7 @@ type CreateDeploymentConfig struct {
 	ContainerPort            int
 	RequestedMemoryMB        float64
 	RequestedCPU             float64
+	RequestedGPU             float64
 	AdditionalContainerPorts []components.ContainerPort
 	Env                      []components.DeploymentConfigV3Env
 	DeploymentTag            string
@@ -490,6 +503,7 @@ func (c *CreateDeploymentConfig) Load(cmd *cli.Command) error {
 	c.ContainerPort = int(cmd.Int(containerPortFlag.Name))
 	c.RequestedMemoryMB = cmd.Float(requestedMemoryFlag.Name)
 	c.RequestedCPU = cmd.Float(requestedCPUFlag.Name)
+	c.RequestedGPU = cmd.Float(requestedGPUFlag.Name)
 	c.DeploymentTag = cmd.String(deploymentTagFlag.Name)
 
 	addlPorts := cmd.StringSlice(additionalContainerPortsFlag.Name)
@@ -542,6 +556,11 @@ func (c *CreateDeploymentConfig) Merge(latest *components.DeploymentV3) {
 		c.RequestedCPU = latest.RequestedCPU
 	}
 
+	if c.RequestedGPU == 0 {
+		// TODO: add support for final API field when it exists
+		c.RequestedGPU = *latest.ExperimentalRequestedGPU
+	}
+
 	if len(c.AdditionalContainerPorts) == 0 {
 		c.AdditionalContainerPorts = latest.AdditionalContainerPorts
 	}