New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Added support for printing scalar values in Pallas TPU kernels #21370

Closed

copybara-service wants to merge 0 commits into main from test_636268627

copybara-service bot commented May 22, 2024 •

edited

Loading

Added support for printing scalar values in Pallas TPU kernels

The implementation uses the new tpu.log operation in the Mosaic TPU dialect.

Note that

the logging only happens if --xla_tpu_enable_log_recorder is set;
only scalars can be printed;
placeholders only accept i32 arguments at the moment.

copybara-service bot assigned superbobry

copybara-service bot force-pushed the test_636268627 branch 4 times, most recently from f3b61fd to 6e0c9d4 Compare

May 23, 2024 16:37

copybara-service bot closed this

copybara-service bot force-pushed the test_636268627 branch from 6e0c9d4 to daa81e6 Compare

May 23, 2024 17:02

copybara-service bot deleted the test_636268627 branch

May 23, 2024 17:02

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