Skip to content

Replace XLA_COUNTER with TORCH_LAZY_COUNTER - #4208

Merged
JackCaoG merged 12 commits into
masterfrom
alanwaketan/counters
Nov 18, 2022
Merged

Replace XLA_COUNTER with TORCH_LAZY_COUNTER#4208
JackCaoG merged 12 commits into
masterfrom
alanwaketan/counters

Conversation

@alanwaketan

@alanwaketan alanwaketan commented Nov 17, 2022

Copy link
Copy Markdown
Collaborator

Summary:
This pull request replaces most of XLA_COUNTER with the pytorch/xla lib with upstream TORCH_LAZY_COUNTER. However, XLA_COUNTER is still kept to be used in the xla_client lib.

Test Plan:
python test/test_metrics.py MetricsTest.test_metrics_report
and CI.

@alanwaketan
alanwaketan marked this pull request as draft November 17, 2022 03:52
@alanwaketan alanwaketan self-assigned this Nov 17, 2022
@alanwaketan alanwaketan added the tracing Lazy Tensor tracing label Nov 17, 2022
@alanwaketan alanwaketan changed the title WIP: Replace XLA_COUNTER with TORCH_LAZY_COUNTER Replace XLA_COUNTER with TORCH_LAZY_COUNTER Nov 17, 2022
@alanwaketan
alanwaketan marked this pull request as ready for review November 17, 2022 06:10
@alanwaketan
alanwaketan requested a review from JackCaoG November 17, 2022 06:10

@JackCaoG JackCaoG left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

Thanks!

@alanwaketan

Copy link
Copy Markdown
Collaborator Author

@JackCaoG Will the REMOVE_TORCH_PIN label remove the pin before merging?

@JackCaoG

Copy link
Copy Markdown
Collaborator

lol nope, it is just a reminder for yourself. you can delete the pin and merge without waiting for CI

@JackCaoG
JackCaoG merged commit 788001c into master Nov 18, 2022
@alanwaketan
alanwaketan deleted the alanwaketan/counters branch December 2, 2022 20:29
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

REMOVE_TORCH_PIN tracing Lazy Tensor tracing

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants