xla
[PT/XLA] Fix timing of ExecuteComputation metrics.
#4327

Merged

[PT/XLA] Fix timing of ExecuteComputation metrics. #4327

stgpetrovic merged 6 commits into master from pjrt-execute-metrics

stgpetrovic force pushed from 6a6763dc to 4e38da1b 3 years ago

stgpetrovic force pushed from fa43570c to d4352dfb 3 years ago

stgpetrovic force pushed from d4352dfb to 2bcbf25a 3 years ago

stgpetrovic force pushed from 2bcbf25a to 5b189fba 3 years ago

stgpetrovic force pushed from 5b189fba to 014272cc 3 years ago

stgpetrovic force pushed from 014272cc to 35a71c78 3 years ago

stgpetrovic force pushed from 35a71c78 to dbb1ebc9 3 years ago

will-cromar approved these changes on 2022-12-21

JackCaoG commented on 2022-12-21

[PT/XLA] Fix timing of ExecuteComputation metrics.

0932c993

Fix comment typo.

206b925e

Add test case.

dd0e9649

ExecuteMetric semantics is max(ExecuteSharded, ExecuteComputation)

0889393e

Suppress execute time metric test on gpu

e02d6c3e

Describe the code a bit better.

7b1a0c9d

stgpetrovic force pushed from eedd5d29 to 7b1a0c9d 3 years ago

stgpetrovic merged afe7bb6b into master 3 years ago

stgpetrovic deleted the pjrt-execute-metrics branch 3 years ago

Reviewers

will-cromar

JackCaoG

Assignees

No one assigned

Labels

None yet

Milestone

No milestone