"mounts": {"/tmp": {"kind": "tmp"} },
"output_path": "/tmp",
"priority": 1,
- "runtime_constraints": {"cuda_device_count": 2},
+ "runtime_constraints": {"cuda": {"device_count": 2}},
"state": "Locked",
"output_storage_classes": ["foo", "bar"]
}`, nil, 0, func() {
"mounts": {"/tmp": {"kind": "tmp"} },
"output_path": "/tmp",
"priority": 1,
- "runtime_constraints": {"cuda_hardware_capability": "foo"},
+ "runtime_constraints": {"cuda": {"hardware_capability": "foo"}},
"state": "Locked",
"output_storage_classes": ["foo", "bar"]
}`, nil, 0, func() {
fmt.Fprintln(s.executor.created.Stdout, "ok")
})
- c.Check(s.executor.created.CUDADeviceCount, Equals, 1)
+ c.Check(s.executor.created.CUDADeviceCount, Equals, 0)
}
func (s *TestSuite) TestStopOnSignal(c *C) {
v = runtime_constraints['cuda'][k]
if !v.is_a?(String) || (runtime_constraints['cuda']['device_count'] > 0 && v.to_f == 0.0)
errors.add(:runtime_constraints,
- "[cuda.#{k}]=#{v.inspect} must be a string in format 'X.Y' version")
+ "[cuda.#{k}]=#{v.inspect} must be a string in format 'X.Y'")
end
end
end