diff options
| author | aiordache <anca.iordache@docker.com> | 2020-10-23 11:45:40 +0200 |
|---|---|---|
| committer | Nicolas De loof <nicolas.deloof@gmail.com> | 2020-11-17 13:34:58 +0100 |
| commit | 854c003359bd07d0d3ca137d7a08509cfeab0436 (patch) | |
| tree | 9db6e134753b44d3c5883d050e43b64444d2be5f /compose/service.py | |
| parent | 843621dfb8ad36ec9301045117630cebac1d917c (diff) | |
Implement device requests for GPU support
Signed-off-by: aiordache <anca.iordache@docker.com>
Diffstat (limited to 'compose/service.py')
| -rw-r--r-- | compose/service.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/compose/service.py b/compose/service.py index a1a500cb..e00a537c 100644 --- a/compose/service.py +++ b/compose/service.py @@ -77,6 +77,7 @@ HOST_CONFIG_KEYS = [ 'cpuset', 'device_cgroup_rules', 'devices', + 'device_requests', 'dns', 'dns_search', 'dns_opt', @@ -180,6 +181,7 @@ class Service: pid_mode=None, default_platform=None, extra_labels=None, + device_requests=None, **options ): self.name = name @@ -195,6 +197,7 @@ class Service: self.secrets = secrets or [] self.scale_num = scale self.default_platform = default_platform + self.device_requests = device_requests self.options = options self.extra_labels = extra_labels or [] @@ -1016,6 +1019,7 @@ class Service: privileged=options.get('privileged', False), network_mode=self.network_mode.mode, devices=options.get('devices'), + device_requests=self.device_requests, dns=options.get('dns'), dns_opt=options.get('dns_opt'), dns_search=options.get('dns_search'), |