Support torch_xla2 benchmarking using torchbench #7013
zpcore
marked this pull request as ready for review 1 year ago
qihqi
approved these changes
on 2024-05-10
add torch_xla2 benchmark to torchbench test
dfad1d09
minor update
8e63770b
minor update
d58dc0d2
minor update
c76db09e
minor update
d166215d
add JAX_PLATFORMS env
c5d1d751
nit update
6698f379
fix bugs
63154592
nit update
4d59e432
update based on feedback
dcab3851
update torch_xla2 argument
be406049
zpcore
force pushed
from
92165f9d
to
be406049
1 year ago
add extract_jax for torch_xla2
a6ba2c14
fix test
0305dcf3
fix test
d55c86ca
fix test
1609dd5f
fix test
de91758a
fix test
a804a5a1
fix test
07c92aff
fix test
d66baa48
fix test
5cfb47e3
zpcore
merged
9e189350
into master 1 year ago
zpcore
deleted the piz/xla2_bm branch 1 year ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub