Skip to content

Commit

Permalink
Add synaptor workers to easyseg instance
Browse files Browse the repository at this point in the history
Only support CPU and GPU tasks, SEGGRAPH tasks requires a database
already, so unlikely we would use easyseg for it
  • Loading branch information
ranlu committed Feb 13, 2024
1 parent baa60fa commit e56e1f5
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion cloud/google/easyseg_worker.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,9 @@ def GenerateEasysegWorker(context, hostname_manager, hostname_easyseg_worker):
chunkflow_cmd = GenerateCeleryWorkerCommand(docker_image, docker_env+['-p 8793:8793'], queue="gpu", concurrency=2)
abiss_cmd = GenerateCeleryWorkerCommand(docker_image, docker_env, queue="atomic", concurrency=1)
igneous_cmd = GenerateDockerCommand(docker_image, docker_env + ["--restart on-failure",]) + ' ' + "python custom/task_execution.py --queue igneous --concurrency 0 >& /dev/null"
cmd += " & \n".join([oom_canary_cmd, chunkflow_cmd, abiss_cmd, igneous_cmd])
synaptor_cpu_cmd = GenerateCeleryWorkerCommand(docker_image, docker_env, queue="synaptor-cpu", concurrency=1)
synaptor_gpu_cmd = GenerateCeleryWorkerCommand(docker_image, docker_env, queue="synaptor-gpu", concurrency=1)
cmd += " & \n".join([oom_canary_cmd, chunkflow_cmd, abiss_cmd, igneous_cmd, synaptor_cpu_cmd, synaptor_gpu_cmd])

diskType = ZonalComputeUrl(
context.env['project'],
Expand Down

0 comments on commit e56e1f5

Please sign in to comment.