pytorch/xla

Pull Requests Commits

[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

[PERF] Adds epoch to mnist test logs (#4112)

alanwaketan committed 3 years ago

Verified 9db07981

[PERF] Add a util script to analyse perf results (#4107)

alanwaketan committed 3 years ago

Verified 9a618c88

Set default xla_gpu_force_compilation_parallelism=8 (#4105)

ymwangg committed 3 years ago

Verified ede67185

Add ConvertXlaShapeToLazy helper function (#4081)

wonjoo-wj committed 3 years ago

Verified bb260ac5

Scope test (#4098)

JackCaoG committed 3 years ago

Verified e4e88060

[SPMD][PoC] compile & execute with PjRt (#3684)

yeounoh committed 3 years ago

Verified 0f23514b

Make sure PjRtData is not empty before accessing opaque handle (#4092)

will-cromar committed 3 years ago

Verified e1f5a496

Fix build script (#4096)

JackCaoG committed 3 years ago

Verified 12801878

implement XLASymIntNodeImpl::wrap (#4076)

Krovatkin committed 3 years ago

Verified 7c0d2993

Use PjRt GPU client (#4038)

jonb377 committed 3 years ago

Verified c58480a0

Add an equality node (#4079)

Krovatkin committed 3 years ago

Verified c5ff2809

Adding bool operator and a test (#4080)

Krovatkin committed 3 years ago

Verified 3db9008f

[DDP] Add a test case to test a larger model (#4085)

alanwaketan committed 3 years ago

Verified b9a32d3d

Newer Older