Skip to content

Conversation

@misiugodfrey
Copy link
Contributor

We want to be able to dynamically generate docker services for multiple workers that use separately assigned gpus.

This PR does so by replacing the docker-compose.native-gpu.yml file with a jinja template that gets expanded via python script when start_native_gpu_presto.sh is called based on the --num-workers (-w) parameter is called. If the option left unspecified, it continues to generate a presto-native-worker-gpu with access to all gpus as normal.

@@ -0,0 +1,50 @@
x-presto-native-worker-gpu: &gpu_worker_base
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

services starting with x- are considered "Extensions" and are ignored when bringing up services. This one is uses exclusively as an anchor.

@karthikeyann
Copy link
Contributor

karthikeyann commented Jan 10, 2026

Do a clean build. Please delete all dockers images and perform multi-node images build and run once. Fix accordingly based on failures.

IIUC, presto-native-worker-gpu image is not built during multi-node build because the recipe has no service presto-native-worker-gpu during multi-node build. I had to build without multi-gpu and then build the multi-gpu again.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants