Skip to content

Commit

Permalink
Fix number of nodes in CPUs. (#96)
Browse files Browse the repository at this point in the history
  • Loading branch information
RoshaniN authored Mar 26, 2024
1 parent 9a964d9 commit faef9a1
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions xpk.py
Original file line number Diff line number Diff line change
Expand Up @@ -2288,6 +2288,9 @@ def run_gke_node_pool_create_command(args, system) -> int:
f' --additional-node-network network={args.cluster}-net-4,subnetwork={args.cluster}-sub-4'
' --no-enable-autoupgrade --scopes="https://www.googleapis.com/auth/cloud-platform"'
)
elif system.accelerator_type == AcceleratorType['CPU']:
command += (f' --num-nodes={system.vms_per_slice}')
command += (' --scopes=storage-full,gke-default')

if _SERVICE_ACCOUNT_FEATURE_FLAG:
service_account_name = get_service_account_name(args)
Expand Down

0 comments on commit faef9a1

Please sign in to comment.