xla
Support torch_xla2 benchmarking using torchbench
#7013
Merged

Support torch_xla2 benchmarking using torchbench #7013

zpcore merged 20 commits into master from piz/xla2_bm
zpcore
zpcore zpcore requested a review from qihqi qihqi 1 year ago
zpcore zpcore requested a review from ysiraichi ysiraichi 1 year ago
zpcore zpcore requested a review from vanbasten23 vanbasten23 1 year ago
zpcore zpcore requested a review from will-cromar will-cromar 1 year ago
zpcore zpcore marked this pull request as ready for review 1 year ago
ysiraichi
ysiraichi commented on 2024-05-02
vanbasten23
vanbasten23 commented on 2024-05-02
vanbasten23
vanbasten23 commented on 2024-05-02
ysiraichi
ysiraichi commented on 2024-05-06
qihqi
qihqi approved these changes on 2024-05-10
zpcore add torch_xla2 benchmark to torchbench test
dfad1d09
zpcore minor update
8e63770b
zpcore minor update
d58dc0d2
zpcore minor update
c76db09e
zpcore minor update
d166215d
zpcore add JAX_PLATFORMS env
c5d1d751
zpcore nit update
6698f379
zpcore fix bugs
63154592
zpcore nit update
4d59e432
zpcore update based on feedback
dcab3851
zpcore update torch_xla2 argument
be406049
zpcore zpcore force pushed from 92165f9d to be406049 1 year ago
zpcore add extract_jax for torch_xla2
a6ba2c14
zpcore fix test
0305dcf3
zpcore fix test
d55c86ca
zpcore fix test
1609dd5f
zpcore fix test
de91758a
zpcore fix test
a804a5a1
zpcore fix test
07c92aff
zpcore fix test
d66baa48
zpcore fix test
5cfb47e3
zpcore zpcore merged 9e189350 into master 1 year ago
zpcore zpcore deleted the piz/xla2_bm branch 1 year ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone