Restrict Pathways unified debugging logs to just first worker

AI-Hypercomputer / xpk

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.

Apache License 2.0

81 stars 23 forks source link

Restrict Pathways unified debugging logs to just first worker #148

Closed SujeethJinesh closed 5 months ago

SujeethJinesh commented 5 months ago

Fixes / Features

Restrict the Pathways unified debugging view to only the first worker's logs.

Testing / Documentation

Tested manually.

[ y ] Tests pass
[ y ] Appropriate changes to documentation are included in the PR