pytorch/xla

Pull Requests Commits

Minor clean up for expand_symint

JackCaoG committed 3 years ago

d7a6836a

Support `expand.symint` with real symbolic symint (#3889)

miladm committed 3 years ago

Verified 31825f3d

Remove `==` constraint from `typing_extensions` in build (#4031)

will-cromar committed 3 years ago

Verified bcb212c6

Revert "Revert "Ceil forr round trunc int (#3913)" (#4019)" (#4022)

JackCaoG committed 3 years ago

Verified cd88d42d

Add Arm Compute Library (ACL) build option for torch-xla (#3542)

snadampal committed 3 years ago

Verified d3fad53b

Fix equation parsing for fallback logic (#4027)

steventk-g committed 3 years ago

Verified 5c4382de

Rename `pjrt.run_multiprocess` to `pjrt._run_multiprocess`. (#4025)

will-cromar committed 3 years ago

Verified 5bac0678

Update libtpu to align with latest TF pin update (#4030)

wonjoo-wj committed 3 years ago

Verified 4226edb6

Update TensorFlow pin (08-2022) (#3922)

wonjoo-wj committed 3 years ago

Verified 5aaab8f6

Codegen HardshrinkBackward and SoftshrinkBackward (#4002)

vanbasten23 committed 3 years ago

Verified ab15402f

Fixing a race where processes other than ones saving can remove the chkpt directory (#4007)

amithrm committed 3 years ago

Verified 0d316f3f

Cache dynamic ir result (#4008)

JackCaoG committed 3 years ago

Verified 5f45b9fe

Fix warning (#4021)

JackCaoG committed 3 years ago

Verified acbb24aa

[PJRT] Flatten output of `pjrt.run_multiprocess` (#3996)

will-cromar committed 3 years ago

Verified eecf89a2

Add PjRt check for test_too_many_parameter test (#4020)

wonjoo-wj committed 3 years ago

Verified b3f79ccf

Revert "Ceil forr round trunc int (#3913)" (#4019)

JackCaoG committed 3 years ago

Verified 06f9ccd1

Ceil forr round trunc int (#3913)

JackCaoG committed 3 years ago

Verified 494e20bb

chain_matmul is removed from PyTorch; it was deprecated since 1.9 (#4017)

IvanYashchuk committed 3 years ago

Verified f020286b

Fix gcc-9 build error: duplicate ‘const’ (#4016)

byronyi committed 3 years ago

Verified a3376130

Add upsample_nearest/bilinear implementation for CPU and GPU (#3990)

ymwangg committed 3 years ago

Verified 307af431

Op lowering for Einsum (#3843)

steventk-g committed 3 years ago

Verified aa735098

Update test name for index_reduce tests (#4010)

wonjoo-wj committed 3 years ago

Verified 54583a3a

Codegen hardshrink (#3999)

vanbasten23 committed 3 years ago

Verified a1691bdd

sym_strides patch (#4009)

ezyang committed 3 years ago

Verified 97686a60

DimCast Helpers (#4006)

Krovatkin committed 3 years ago

Verified 44104e7e

allow FSDP wrapping and sharding over modules on CPU devices (#3992)

ronghanghu committed 3 years ago

Verified 6a133b87

switch everything to XLASymIntNodeImpl (#4001)

Krovatkin committed 3 years ago

Verified a014e031

Add ComputationClient metrics to PjRtComputationClient (#3984)

jonb377 committed 3 years ago

Verified a0893579

Implement getDynamicValue for SizeNode (#3988)

JackCaoG committed 3 years ago

Verified c6cff0d9

Expand static symint test (#4004)

JackCaoG committed 3 years ago

Verified 90320ce0

Older