xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerators such as TPUs and GPUs on GKE.
Add argument termination_grace_period_seconds to provide maximum wait time for a workload pod to wrap up after a disruption event or deletion request.
Testing / Documentation
Testing details.
Created a cluster with 2 slices of v4-8 and a workload using -tgps and confirmed the user provided terminationGracePeriodSeconds value in workload pod job yaml.
[ y ] Tests pass
[ y ] Appropriate changes to documentation are included in the PR
Fixes / Features
termination_grace_period_secondsto provide maximum wait time for a workload pod to wrap up after a disruption event or deletion request.Testing / Documentation
Testing details. Created a cluster with 2 slices of v4-8 and a workload using
-tgpsand confirmed the user providedterminationGracePeriodSecondsvalue in workload pod job yaml.[ y ] Tests pass [ y ] Appropriate changes to documentation are included in the PR