pytorch/xla

Pull Requests Commits

Add mark_tensor variant taking torch.Tensor id (#6132)

lsy323 committed 1 year ago

Verified d6d8d477

Revert "Initialize compilation env hash" (#6147)

jonb377 committed 1 year ago

Verified 6aaca152

Initialize compilation env hash (#6140)

jonb377 committed 1 year ago

Verified c4f8772d

Do manual type conversion for isnan op (#6110)

wonjoo-wj committed 1 year ago

Verified 5c12670a

Uncomment test_aten_isinf_2 at test_core_aten_ops.py (#6105)

wonjoo-wj committed 1 year ago

Verified fe7e1fde

Revert "Add comments to _is_available and raise exceptions in unimple… (#6141)

frgossen committed 1 year ago

Verified ddc2e34a

Add flag for equal_nan in test_core_aten_ops.py (#6104)

wonjoo-wj committed 1 year ago

Verified e326ed29

Do not fail on lack of default precision set. (#6139)

golechwierowicz committed 1 year ago

Verified 6b1344e6

ci: install pandas (#6137)

cota committed 1 year ago

Verified a2f80e46

benchmarks/aggregate: remove timestamp from "latest" title (#6138)

cota committed 1 year ago

Verified 9f9c9088

Add comments to _is_available and raise exceptions in unimplemented cases (#6118)

frgossen committed 1 year ago

Verified 491cbed7

Revert "Fix model initialization. (#6076)" (#6134)

frgossen committed 1 year ago

Verified 788e1b5e

Apply default precision from bench models to XLA (#6133)

golechwierowicz committed 1 year ago

Verified a0856820

Re-land experiment runner tests (#6113)

frgossen committed 1 year ago

Verified 3bbafc2a

Enforce successful imports and cause failure if not set up correctly (#6119)

frgossen committed 1 year ago

Verified ad02e793

Add log level parsing to argparse (#6126)

frgossen committed 1 year ago

Verified 33990ee9

benchmarks/aggregate: fix in-place map (#6136)

cota committed 1 year ago

Verified 789365cc

Remove unused experiment name, which was always `run_all` (#6135)

frgossen committed 1 year ago

Verified a29ff4ec

Deprecate explicit shape promotion in PyTorch/XLA (#6121)

sdasgup3 committed 1 year ago

Verified 270e1bc7

Handle negative dim for Diagonal Scatter (#6123)

JackCaoG committed 1 year ago

Verified 983997c7

Fix bad error message when `PjRtComputationClient` throws exception (#5946)

will-cromar committed 1 year ago

Verified 280ca1d8

benchmarks/patches: fix corrupt patch file error (#6124)

cota committed 1 year ago

Verified fc713b78

Uncomment test_aten_scatter_add_1 (#6102)

wonjoo-wj committed 1 year ago

Verified 2d359a36

[SPMD] Allow mark_shard scalar tensors (#6101)

alanwaketan committed 1 year ago

Verified 45a9e0e9

Remove flaky test (#6116)

jonb377 committed 1 year ago

Verified dd50c364

Fix model initialization. (#6076)

ysiraichi committed 1 year ago

Verified 7786d5df

Handled lifted constants in torch export. (#6111)

qihqi committed 1 year ago

Verified d8d7e58b

[SPMD] Add apply_backward_optimization_barrier (#6097)

alanwaketan committed 1 year ago

Verified 07540f21

Fix bug for low-precision ranges (#5636)

jonb377 committed 1 year ago

Verified eb3c446f

Add Pattern Boundary Marking API (#5930)

lsy323 committed 1 year ago

Verified 8243b502

Newer Older