fix: add utility capability for local_docker (#906)

Alexander Jipa · web-flow · commit ec8d0e855170 · 2024-05-07T16:17:30.000-07:00
Differential Revision: D57034038 Pull Request resolved: #907
diff --git a/torchx/schedulers/docker_scheduler.py b/torchx/schedulers/docker_scheduler.py
@@ -314,7 +314,7 @@ def _submit_dryrun(self, app: AppDef, cfg: DockerOpts) -> AppDryRunInfo[DockerJo
                     c.kwargs["device_requests"] = [
                         DeviceRequest(
                             count=resource.gpu,
-                            capabilities=[["compute"]],
+                            capabilities=[["compute", "utility"]],
                         )
                     ]
                 req.containers.append(c)
diff --git a/torchx/schedulers/test/docker_scheduler_test.py b/torchx/schedulers/test/docker_scheduler_test.py
@@ -91,7 +91,7 @@ def test_submit_dryrun(self) -> None:
                         "device_requests": [
                             DeviceRequest(
                                 count=4,
-                                capabilities=[["compute"]],
+                                capabilities=[["compute", "utility"]],
                             )
                         ],
                         "devices": [

Original file line number	Diff line number	Diff line change
`@@ -314,7 +314,7 @@ def _submit_dryrun(self, app: AppDef, cfg: DockerOpts) -> AppDryRunInfo[DockerJo`
`314`	`314`	`c.kwargs["device_requests"] = [`
`315`	`315`	`DeviceRequest(`
`316`	`316`	`count=resource.gpu,`
`317`		`- capabilities=[["compute"]],`
	`317`	`+ capabilities=[["compute", "utility"]],`
`318`	`318`	`)`
`319`	`319`	`]`
`320`	`320`	`req.containers.append(c)`
Original file line number	Diff line number	Diff line change
`@@ -91,7 +91,7 @@ def test_submit_dryrun(self) -> None:`
`91`	`91`	`"device_requests": [`
`92`	`92`	`DeviceRequest(`
`93`	`93`	`count=4,`
`94`		`- capabilities=[["compute"]],`
	`94`	`+ capabilities=[["compute", "utility"]],`
`95`	`95`	`)`
`96`	`96`	`],`
`97`	`97`	`"devices": [`