pytorch/xla

Pull Requests Commits

vanbasten23 committed 3 years ago

bedc3ed2

Added failing tests for mixed data type cases. All tests pass.

vanbasten23 committed 3 years ago

3a0c6426

fixed a typo in yaml file.

vanbasten23 committed 3 years ago

640ea5b3

vanbasten23 committed 3 years ago

43934e0e

cleaned up a bit.

vanbasten23 committed 3 years ago

fb03f3a0

vanbasten23 committed 3 years ago

238022dc

Codegen HardshrinkBackward and SoftshrinkBackward.

vanbasten23 committed 3 years ago

90f842ae

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

[SPMD][PoC] InputHandler for input sharding (#3963)

yeounoh committed 3 years ago

Verified ffa5b340

Disable python deployment test (#4003)

JackCaoG committed 3 years ago

Verified 2c778496

Make alias return a fresh TensorImpl. (#3998)

ezyang committed 3 years ago

Verified 15851f50

switching to XLASymIntNodeImpl + test : part I (#3979)

Krovatkin committed 3 years ago

Verified a69541e7

Disable test_assertRaisesRegex_ignore_msg_non_native_device (#4000)

JackCaoG committed 3 years ago

Verified 1ff5c8ba

Fix `time` statements in `common.sh` (#3997)

will-cromar committed 3 years ago

Verified b37a259a

Add upper_bound_ and is_symbolic_ to SizeNode (#3958)

JackCaoG committed 3 years ago

Verified a58a99c5

Don't load libtpu on deployment_tests (#3991)

JackCaoG committed 3 years ago

Verified eed0a216

Make SizeNode to use XlaShape and add test (#3954)

JackCaoG committed 3 years ago

Verified e9b8d435

Making sure XLA_HLO_DEBUG populate the scope metadata (#3985)

JackCaoG committed 3 years ago

Verified 0aed7b64

Fix special scalar handling for addcdiv and addcmul (#3953)

ymwangg committed 3 years ago

Verified 88d934e3

Set three_fry as the default RNG for GPU (#3951)

ymwangg committed 3 years ago

Verified 2d877167

Create README.md for CircleCI (#3916)

wonjoo-wj committed 3 years ago

Verified 2bca9530

Older