Skip to content

Commit

Permalink
GCE GPU workers cannot migrate
Browse files Browse the repository at this point in the history
They have to be terminated on host maintaince events.
  • Loading branch information
ranlu committed Mar 3, 2024
1 parent d820001 commit 27b96ae
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions cloud/google/workers.py
Original file line number Diff line number Diff line change
Expand Up @@ -158,6 +158,8 @@ def GenerateWorkers(context, hostname_manager, hostname_nfs_server, worker):
'acceleratorCount': accelerator_count,
'acceleratorType': worker['gpuWorkerAcceleratorType'],
}]
if not worker.get('preemptible', False):
instance_template['scheduling']['onHostMaintenance'] = 'TERMINATE'

if worker['type'] == "atomic":
instance_template['advancedMachineFeatures'] = {
Expand Down

0 comments on commit 27b96ae

Please sign in to comment.