[r2.3] backport PR#6719#6754

Closed

yeounoh wants to merge 13 commits intomasterfrom port_pr6719

+1,228-349

Commits on Mar 13, 2024

Add IsVirtualDeivce() and refactor

yeounoh
authored and
Siyuan Liu
committed
Backport #6704 to r2.3 (#6736 )
lsy323
authored
fix csv output error (#6662 ) (#6738 )

zpcore
and
root
authored
Backport #6677 to r2.3 (#6737 )

lsy323
and
Siyuan Liu
authored
Backport #6730 (#6739 )
lsy323
authored
[r2.3] backport #6723 (#6734 )
yeounoh
authored
add torch pin to 2.3 release branch (#6735 )

lsy323
and
Siyuan Liu
authored
Update torch_xla_cuda_plugin version for r2.3 (#6729 )
will-cromar
authored
Backport Bump JAX version in setup.py (#6740 ) (#6741 )

lsy323
and
will-cromar
authored
[Pallas] Introduce make_kernel_from_pallas (#6713 ) (#6742 )

lsy323
and
alanwaketan
authored
Properly lower aten::sub and aten::rsub (#6668 ) (#6743 )

lsy323
and
anw90
authored
Properly lower add and mul (#6731 ) (#6744 )

lsy323
and
wonjoolee95
authored

Commits on Mar 14, 2024

[SPMD] auto-sharding PoC (#6719 )
yeounoh
committed