pytorch/xla

Pull Requests Commits

update patchs again

ManfeiBai committed 2 years ago

0f128d8a

fix conflict from Dec2 with Dec15

ManfeiBai committed 2 years ago

c5e6a504

Merge branch 'master' of https://github.com/pytorch/xla

ManfeiBai committed 2 years ago

b65dafd2

Revert "Revert "Use C10_AS_INTARRAYREF_SLOW (#4325)" (#4346)" (#4360)

JackCaoG committed 2 years ago

Verified 547ddf79

ManfeiBai committed 2 years ago

470c8fbc

[PJRT] Fix TPU local ordinal calculation (#4334)

will-cromar committed 2 years ago

Verified 562a80d9

Merge branch 'master' of https://github.com/pytorch/xla

ManfeiBai committed 2 years ago

61c29d60

ManfeiBai committed 2 years ago

bbe9aea7

Temporarily remove MKL related tests (#4345)

steventk-g committed 2 years ago

Verified 09a6e7f1

Store Terraform state in shared GCS bucket. (#4323)

will-cromar committed 2 years ago

Verified fc500b47

udpate pytorch to the latest

ManfeiBai committed 2 years ago

61daee52

Revert "Use C10_AS_INTARRAYREF_SLOW (#4325)" (#4346)

JackCaoG committed 2 years ago

Verified 45542f75

ManfeiBai committed 2 years ago

9e5e7601

Use sharding hash (#4287)

steventk-g committed 2 years ago

Verified 68e5f1be

Add sharding hash to IR nodes (#4286)

steventk-g committed 2 years ago

Verified 98b788e2

Promote int to float for tan operation (consistent with Pytorch) (#4333)

steventk-g committed 2 years ago

Verified 4f375efc

ManfeiBai committed 2 years ago

5059672f

[SPMD] Use d2d replication via CopyToDevice (#4336)

yeounoh committed 2 years ago

Verified 5eaea4c8

Add test to verify virtual device usage metrics (#4330)

steventk-g committed 2 years ago

Verified 3f015284

fix maxpool2d for XLA dynamo tracing (#4276)

bdhirsh committed 2 years ago

Verified 4013c57b

Update xla_base tag to v0.5 (#4342)

yeounoh committed 2 years ago

Verified 4d14622c

Revert "Revert "Install sympy to ci base docker image (#4337)" (#4339)" (#4340)

JackCaoG committed 2 years ago

Verified 823dd005

ManfeiBai committed 2 years ago

bb767237

ManfeiBai committed 2 years ago

b52f278b

Revert "Install sympy to ci base docker image (#4337)" (#4339)

miladm committed 2 years ago

Verified 40bd7d6f

Install sympy to ci base docker image (#4337)

yeounoh committed 2 years ago

Verified a0fa2894

ManfeiBai committed 2 years ago

846574b0

ManfeiBai committed 2 years ago

a4f8d6fe

add xla_util.h change

ManfeiBai committed 2 years ago

df988b52

ManfeiBai committed 2 years ago

55294c68

Older