pytorch/xla

Pull Requests Commits

move metrics test to a new file

JackCaoG committed 2 years ago

cdb5d174

Add a api to clear counter and metrics

JackCaoG committed 2 years ago

f1257ac5

Add ConvertXlaShapeToLazy helper function (#4081)

wonjoolee95 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

Codegen Softshrink (#4061)

vanbasten23 committed 2 years ago

Verified 039629c0

Move einsum to failing tests (#4053)

steventk-g committed 2 years ago

Verified 944ace9a

[Distributed] Companion fix after pytorch PR73164 (#3983)

alanwaketan committed 2 years ago

Verified 3edfe4b0

[TorchPin] Update the documentation (#4064)

alanwaketan committed 2 years ago

Verified 74b03a82

Adding SizeConstant (#3994)

Krovatkin committed 2 years ago

Verified ca896bc7

Add 'Add' and 'FloorDiv' to XLASymIntNodeImpl (#4057)

vanbasten23 committed 2 years ago

Verified e515d31d

Fix `ExpectCounterNotChanged` to correctly account for `_symint` (#4060)

miladm committed 2 years ago

Verified 66666bcd

add aarch64 build scripts for torch and xla wheel building (#3660)

snadampal committed 2 years ago

Verified 3eddd035

Revert "Update amax/amin/norm signatures with int[*]? dim (#4035)" (#4059)

JackCaoG committed 2 years ago

Verified 49aed3a1

Older