xla
Add profiler traces to `PjRtComputationClient`
#3866

Merged

Add profiler traces to `PjRtComputationClient` #3866

will-cromar merged 7 commits into master from pjrt-profiler

Add traces to PjRtComputationClient

fb1adaf0

will-cromar added runtime

will-cromar force pushed from abd4f793 to 1283e45e 3 years ago

Add simple profiler test

cc6aaace

will-cromar force pushed from 1283e45e to cc6aaace 3 years ago

Add to CI tests

bd003325

Make compatible with TPU

f8bce8da

will-cromar requested a review from

darisoy 3 years ago

will-cromar requested a review from

JackCaoG 3 years ago

will-cromar marked this pull request as ready for review 3 years ago

will-cromar commented on 2022-08-10

darisoy approved these changes on 2022-08-11

Make test less flaky

5ae035de

whitespace

7bf2118d

Remove PJRT profiler test from run_tests.sh

fa05f38c

JackCaoG approved these changes on 2022-08-12

miladm approved these changes on 2022-08-13

will-cromar merged 48bf0fde into master 3 years ago

Reviewers

miladm

JackCaoG

darisoy

Assignees

No one assigned

Labels

runtime

Milestone

No milestone