Skip to content
This repository was archived by the owner on Mar 3, 2026. It is now read-only.

Add TPU verbose logging flags#122

Open
tengyifei wants to merge 1 commit intomainfrom
tengyifei-patch-1
Open

Add TPU verbose logging flags#122
tengyifei wants to merge 1 commit intomainfrom
tengyifei-patch-1

Conversation

@tengyifei
Copy link
Contributor

No description provided.

@tengyifei tengyifei marked this pull request as ready for review February 26, 2025 01:36
@tengyifei tengyifei requested review from bhavya01 and zpcore February 26, 2025 02:02
@tengyifei tengyifei enabled auto-merge (squash) February 26, 2025 02:02
@bhavya01
Copy link
Collaborator

Is it possible to show examples of what kind of logs do these variables help with in the PR description?

Comment on lines +52 to +55
os.environ["TPU_VMODULE"] = "slice_configuration=1,real_program_continuator=1"
os.environ["TPU_STDERR_LOG_LEVEL"] = "0"
os.environ["TPU_MIN_LOG_LEVEL"] = "0"
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "0"
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It might be helpful to have in code information on what these flags individually do

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Did you forward the env for xpk command?

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants