pytorch/xla

Pull Requests Commits

Enable cloud cache

wonjoo-wj committed 2 years ago

b8295c0c

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 2 years ago

Verified 3434c3ea

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

JackCaoG committed 2 years ago

Verified 1812b1d1

Test fix for Bernoulli (#4118)

ezyang committed 2 years ago

Verified caad8b7a

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

JackCaoG committed 2 years ago

Verified 30d5f23a

reduce the overhead incurred in FetchTensorData (#4011)

aws-rhsoln committed 2 years ago

Verified d3b1a7db

[LTC] Adopt the extensible LazyGraphExecutor (#4106)

alanwaketan committed 2 years ago

Verified 0cb29daa

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

JackCaoG committed 2 years ago

Verified f0020b49

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

alanwaketan committed 2 years ago

Verified 9db07981

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

alanwaketan committed 2 years ago

Verified 9a618c88

Set default xla_gpu_force_compilation_parallelism=8 (#4105)

ymwangg committed 2 years ago

Verified ede67185

Add ConvertXlaShapeToLazy helper function (#4081)

wonjoo-wj committed 2 years ago

Verified bb260ac5

Scope test (#4098)

JackCaoG committed 2 years ago

Verified e4e88060

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

yeounoh committed 2 years ago

Verified 0f23514b

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

will-cromar committed 2 years ago

Verified e1f5a496

Fix build script (#4096)

JackCaoG committed 2 years ago

Verified 12801878

implement XLASymIntNodeImpl::wrap (#4076)

Krovatkin committed 2 years ago

Verified 7c0d2993

Use PjRt GPU client (#4038)

jonb377 committed 2 years ago

Verified c58480a0

Add an equality node (#4079)

Krovatkin committed 2 years ago

Verified c5ff2809

Adding bool operator and a test (#4080)

Krovatkin committed 2 years ago

Verified 3db9008f

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

alanwaketan committed 2 years ago

Verified b9a32d3d

Make sure XrtData is not empty before accessing handle (#4090)

JackCaoG committed 2 years ago

Verified 4a267b43

[PJRT] Enable PJRT C API option (#4077)

will-cromar committed 2 years ago

Verified afaab1f6

[aarch64] build torch library with mkldnn+acl backend (#4066)

snadampal committed 2 years ago

Verified 79c433a0

Implement str() for XLASymIntNodeImpl (#4082)

vanbasten23 committed 2 years ago

Verified a2e995a0

Update PyTorch/XLA version to 1.14 (#4088)

JackCaoG committed 2 years ago

Verified 2467c05c

Rename to fromIntArrayRefSlow (#4071)

ezyang committed 2 years ago

Verified 8332ac90

TF/libtpu Pin Update (#4036)

JackCaoG committed 2 years ago

Verified 90341158

enable test/nn/test_embedding.py (#4063)

kshitij12345 committed 2 years ago

Verified 1874859d

Track diagonal_backward change to symint (#4067)

ezyang committed 2 years ago

Verified b8bf944e

Older