pytorch/xla

Pull Requests Commits

avizon-aws committed 1 year ago

da71b184

fixed formatting

avizon-aws committed 1 year ago

67882907

added softmax to autocast policy

avizon-aws committed 1 year ago

ad174034

When modifying IR node, make sure to not lose the read_only bit (#8505)

mcuiaws committed 1 year ago

Verified e4a9e125

Remove SPMD logging to indicate that we're using virtual device (#8501)

bhavya01 committed 1 year ago

Verified 6ca816d7

xm.save() should not set sync_xla_data=True when sync'ing. (#8484)

mcuiaws committed 1 year ago

Verified d3ed9827

Fix a DDP graph capture issue (#8489)

Chengji Yao committed 1 year ago

Verified b1869a8a

Add xm.xla_device_kind() to return XLA device kind string. (#8493)

mcuiaws committed 1 year ago

Verified 0121444c

Compute and hash buffer_donor_indices for step marker (#8467)

mcuiaws committed 1 year ago

Verified 01548502

Migrate from capture_pre_autograd_graph to export_for_training (#8398)

yushangdi committed 1 year ago

Verified b2b890e9

Update nightly versions to 2.7 in README (#8490)

tengyifei committed 1 year ago

Verified 3f915ab8

Add build triggers for 2.6-rc1 (#8485)

tengyifei committed 1 year ago

Verified 8e37d509

Update test_train_spmd_linear_model.py to show clear loss value (#8481)

ManfeiBai committed 1 year ago

Verified c05de860

Update libtpu/xla/jax pin (#8480)

lsy323 committed 1 year ago

Verified 7261c794

Make it easy to switch to nightly libtpu and jax (#8468)

tengyifei committed 1 year ago

Verified 7dd2697a

[AdHoc] Test fix: clear metrics and simplify the replicated counter (#8478)

rpsilva-aws committed 1 year ago

Verified 7ef4a120

Update torch_xla wheels to have the suffix +cxx11 instead of .cxx11 (#8472)

bhavya01 committed 1 year ago

Verified aa56abcc

Register offloading ops as torch ops (#8477)

tengyifei committed 1 year ago

Verified e8720ba6

flash_attention: support also cross attention. (#8427)

dudulightricks committed 1 year ago

Verified 5790092f

[1/N] Use std::string_view (#7297)

cyyever committed 1 year ago

Verified 58be354d

Add autocast support for XlaPatchedLinear (#8421)

aws-nm9 committed 1 year ago

Verified 2021c0fe

[LoweringContext] SPMD propagation (#8471)

rpsilva-aws committed 1 year ago

Verified b068cab9

Modify the deprecation plan for XLA_USE_BF16 (#8474)

rpsilva-aws committed 1 year ago

Verified 18128175

Fix #8475: add libtpu-wheels registry when building docker (#8476)

tengyifei committed 1 year ago

Verified 47f737a7

Add autocast support for einsum (#8420)

aws-nm9 committed 1 year ago

Verified 7b5aca68

Remove Jack and Will from code owner (#8473)

JackCaoG committed 1 year ago

Verified 5c062ea3

Support S32/U32 indices for BWD embedding & Neuron implicit downcast (#8462)

rpsilva-aws committed 1 year ago

Verified 00c0e966

[LoweringContext] Support an optimized parameter mapping for SPMD (#8460)

rpsilva-aws committed 1 year ago

Verified 5d11f660

Improve wheel naming logic in ansible (#8465)

tengyifei committed 1 year ago

Verified cff26e5b

Update libtpu pin to 2024/12/05 / 0.0.5 stable (#8461)

tengyifei committed 1 year ago

Verified 85cf18ac

Older