pytorch/xla

Pull Requests Commits

JackCaoG committed 2 years ago

e4439cf3

fix IsTpuDevice

JackCaoG committed 2 years ago

58363642

fix review comments

JackCaoG committed 2 years ago

ad3bda93

add test for new virtual device usage

JackCaoG committed 2 years ago

511c3892

remove hacks in test_xla_virtual_device.py

JackCaoG committed 2 years ago

95448436

fix spmd device casting issue

JackCaoG committed 2 years ago

e7d460d6

Fix test_shard_hashing

JackCaoG committed 2 years ago

e56d25b4

fix the waitdeviceop for spmd case

JackCaoG committed 2 years ago

132fd6e7

fix test_mark_sharding_2d by not creating placeholder for virtual device

JackCaoG committed 2 years ago

0aba6a5b

fix load shards

JackCaoG committed 2 years ago

bbcb321e

Move all tensors to SPMD:0 C++ device under spmd context

JackCaoG committed 2 years ago

5d1683f0

Fix the error where mark_step does not materalize tensors on SPMD:0 (#5281)

JackCaoG committed 2 years ago

Verified cb5944ae

Make it an error if calling sizes() on a dynamic tensor. (#4998)

vanbasten23 committed 2 years ago

Verified d3779cb8

Add `runtime.host_index` (#5283)

will-cromar committed 2 years ago

Verified 68d577ee

Remove or improve several hardcoded TPU test conditions (#5272)

will-cromar committed 2 years ago

Verified b8617f52

Always do build_ext in python setup.py develop (#5273)

qihqi committed 2 years ago

Verified 71e6e9b0

Dsiable test_simple_model_with_different_input_shape since it is curretnly broken by pytorch (#5282)

JackCaoG committed 2 years ago

Verified b799e206

Fix the error when export_torch_model is given a non-tensor (#5277)

qihqi committed 2 years ago

Verified 23e722f6

Delete unused .so file, and .lds files (#5275)

ManfeiBai committed 2 years ago

Verified ba01cf9a

Clean bazel stuff on distutils clean. (#5274)

stgpetrovic committed 2 years ago

Verified 1842414c

Make TPU detection more robust (#5271)

will-cromar committed 2 years ago

Verified 048f3fa6

[SPMD] Sharding n-d tensor on (n+1)-d Mesh (#5268)

khatwanimohit committed 2 years ago

Verified adc97adc

Suppress debug symbols in OpenXLA code (#5269)

stgpetrovic committed 2 years ago

Verified 625d2d07

Disable Bazel remote cache for forked PR (#5259)

lsy323 committed 2 years ago

Verified 81c5cc07

Enable PjRt Client Compilation with StableHLO (#5233)

lsy323 committed 2 years ago

Verified 905be9d5

Update Sharded graph HLO dumping (#5266)

JackCaoG committed 2 years ago

Verified 0fdc3fb4

Add ToString method for both PjrtData and PjrtShardedData (#5265)

JackCaoG committed 2 years ago

Verified e7ad09ed

[OpenXLA] Migrate to pull XLA from OpenXLA (#5202)

ManfeiBai committed 2 years ago

Verified 6f9cb615

Add XRT nightly builds (#5261)

will-cromar committed 2 years ago

Verified 2e6b9bb7

Skip calling as_strided in empty_strided_symint if the input has dynamic dimensions. (#5239)

vanbasten23 committed 2 years ago

Verified f7237922

Older