pytorch/xla

Pull Requests Commits

[SPMD] Mesh to support custom device order. (#4162)

yeounoh committed 3 years ago

Verified b096c5cf

[LTC] Remove XLAIrBuilder::MakeView (#4154)

alanwaketan committed 3 years ago

Verified f7c0775b

Companion PR for 'SymIntify _copy functionalization kernels (and _copy_out too)' (#4164)

ezyang committed 3 years ago

Verified 79cbdb60

Add some basic Dynamo test (#4159)

JackCaoG committed 3 years ago

Verified 515d53e9

Enable `padded` option for even tensor sharding (#4156)

yeounoh committed 3 years ago

Verified 5ff192ea

Update codegen to use xla::Shape (#4111)

wonjoo-wj committed 3 years ago

Verified 587a6b6d

Refactor argparser to be consistent with torch.distributed (#4149)

muellerzr committed 3 years ago

Verified bde1bc6a

Add short_metrics (#4148)

JackCaoG committed 3 years ago

Verified 1f370695

update docs (#4151)

JackCaoG committed 3 years ago

Verified 9ae3cbbb

[SPMD] Test sharding_spec with clear_sharding, deepcopy (#4144)

yeounoh committed 3 years ago

Verified 590dee5d

Fix clear counter (#4143)

JackCaoG committed 3 years ago

Verified 6675ada5

Update release 1.13 in README file. (#4140)

vanbasten23 committed 3 years ago

Verified 615aa68a

enable nn/convolution tests (#4120)

kshitij12345 committed 3 years ago

Verified ad84ae8a

[DDP] Add a README (#4138)

alanwaketan committed 3 years ago

Verified e0dd3e08

[PERF] Adds DDP mode to imagenet and mnist tests (#4135)

alanwaketan committed 3 years ago

Verified f224b7ff

[SPMD] Use on_device_shape in ExecuteReplicated (#4139)

yeounoh committed 3 years ago

Verified 8f3edf45

Update for unify SymIntNode and SymFloatNode into SymNode (#4132)

ezyang committed 3 years ago

Verified 926905a0

[LTC] Remove IrBuilder trimmed interfaces (#4129)

alanwaketan committed 3 years ago

Verified aaf3ce7d

TF/libtpu Pin Update 10/17/2022 (#4101)

steventk-g committed 3 years ago

Verified 647a3d5a

Dynamo Poc (#4119)

JackCaoG committed 3 years ago

Verified e3a6e3a8

Enable cloud cache (#4123)

wonjoo-wj committed 3 years ago

Verified 79131e9d

Release document for PJRT (#4121)

will-cromar committed 3 years ago

Verified 9d383743

Significantly reduce Docker image size (#4044)

will-cromar committed 3 years ago

Verified 73c2fd06

node.js v18 gets installed by default since 10/25/2022 due to new lts version available. Lock the version to v16 on bionic (#4125)

weiwangmeta committed 3 years ago

Verified 3434c3ea

Disable cloud cache since it is failing on upstream CI (#4122)

JackCaoG committed 3 years ago

Verified 1812b1d1

Test fix for Bernoulli (#4118)

ezyang committed 3 years ago

Verified caad8b7a

Disable fallback on xla::dot int inputs (#4116)

JackCaoG committed 3 years ago

Verified 30d5f23a

reduce the overhead incurred in FetchTensorData (#4011)

aws-rhsoln committed 3 years ago

Verified d3b1a7db

[LTC] Adopt the extensible LazyGraphExecutor (#4106)

alanwaketan committed 3 years ago

Verified 0cb29daa

Add an api to clear counter and metrics (#4109)

JackCaoG committed 3 years ago

Verified f0020b49

Newer Older