X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/f04d5211ed026a4e0cbdca77dad447700eb88772..7865723dc5eee129e7ac269f3495274a13ff70ae:/sdk/go/arvados/container.go

diff --git a/sdk/go/arvados/container.go b/sdk/go/arvados/container.go
index 27afc1a3ab..3510a6db04 100644
--- a/sdk/go/arvados/container.go
+++ b/sdk/go/arvados/container.go
@@ -93,16 +93,20 @@ type Mount struct {
 	GitURL            string      `json:"git_url"`         // only if kind=="git_tree"
 }
 
+type CUDARuntimeConstraints struct {
+	DriverVersion      string `json:"driver_version"`
+	HardwareCapability string `json:"hardware_capability"`
+	DeviceCount        int    `json:"device_count"`
+}
+
 // RuntimeConstraints specify a container's compute resources (RAM,
 // CPU) and network connectivity.
 type RuntimeConstraints struct {
-	API                    bool   `json:"API"`
-	RAM                    int64  `json:"ram"`
-	VCPUs                  int    `json:"vcpus"`
-	KeepCacheRAM           int64  `json:"keep_cache_ram"`
-	CUDADriverVersion      string `json:"cuda_driver_version,omitempty"`
-	CUDAHardwareCapability string `json:"cuda_hardware_capability,omitempty"`
-	CUDADeviceCount        int    `json:"cuda_device_count,omitempty"`
+	API          bool                   `json:"API"`
+	RAM          int64                  `json:"ram"`
+	VCPUs        int                    `json:"vcpus"`
+	KeepCacheRAM int64                  `json:"keep_cache_ram"`
+	CUDA         CUDARuntimeConstraints `json:"cuda"`
 }
 
 // SchedulingParameters specify a container's scheduling parameters