pytorch/xla

Pull Requests Commits

Remove .torch_pin

alanwaketan committed 2 years ago

3379a257

alanwaketan committed 2 years ago

640a5b8a

alanwaketan committed 2 years ago

ab0d638f

Update .torch_pin

alanwaketan committed 2 years ago

5289ad07

[LTC] Replace XLA_VALUE_METRIC with TORCH_LAZY_VALUE_METRIC

alanwaketan committed 2 years ago

8101f573

[LTC] Replace XLA_FN_COUNTER with TORCH_LAZY_FN_COUNTER (#4217)

alanwaketan committed 2 years ago

Verified b969cba3

Companion PR for pytorch/pytorch#89327 (#4242)

Nikita Karetnikov committed 2 years ago

Verified 71c5a005

[SPMD] fixes bugs with AssignIrValue & ExecuteReplicated (#4233)

yeounoh committed 2 years ago

Verified 25cb9975

Optionally bundle libtpu into TPU VM wheel (#4204)

will-cromar committed 2 years ago

Verified f8581559

Add fork start_method support for xmp.spawn (#4236)

darisoy committed 2 years ago

Verified 7c0acfc7

Partially Codegen Baddbmm (#4222)

ManfeiBai committed 2 years ago

Verified 1251261c

Fix a corner case where view tensor update will be ignore by mark_step (#4232)

JackCaoG committed 2 years ago

Verified ba9f0df0

fix casting SymInt to XLASymNodeImpl (#4212)

Krovatkin committed 2 years ago

Verified d471d08e

[SPMD] do add instead of matmul in SPMD unit testing (#4237)

yeounoh committed 2 years ago

Verified 7a4ee34d

Companion PR for pytorch/pytorch#89326 (#4240)

Nikita Karetnikov committed 2 years ago

Verified 281f3696

Disable failing take python op tests on TPU (#4231)

wonjoo-wj committed 2 years ago

Verified 4fdee512

inplace operation within no_grad context for all_gather (#4235)

aws-rhsoln committed 2 years ago

Verified 0e955758

XLA Op Key Registration should only happen once. (#4228)

JackCaoG committed 2 years ago

Verified 98ecca30

model should be on xla device for Dynamo torchxla_trace_once backend (#4205)

JackCaoG committed 2 years ago

Verified 36e2acab

Add const override flag to getstaticValue (#4229)

JackCaoG committed 2 years ago

Verified 6a792ddc

move all patches to different files and add comments (#4225)

JackCaoG committed 2 years ago

Verified bbcea0f2

Add python test for sizeAdd node and implement `int_` for XLASymNodeImpl. (#4218)

vanbasten23 committed 2 years ago

Verified f790bc8a

Introduce virtual device class (#4091)

steventk-g committed 2 years ago

Verified b2bd721c

Replace XLA_COUNTER with TORCH_LAZY_COUNTER (#4208)

alanwaketan committed 2 years ago

Verified 788001c5

Improve XLASynNodeImpl str() format (#4216)

vanbasten23 committed 2 years ago

Verified a46c616e

Add a api to reset and extract RNG Seed (#4196)

JackCaoG committed 2 years ago

Verified 0daf6fe9

Implement `xm.rendezvous` with XLA collective communication for PJRT (#4181)

will-cromar committed 2 years ago

Verified a9eb3459

[SPMD] fix sharding util bugs & implement unit test cases for 5D tile sharding (#4192)

yeounoh committed 2 years ago

Verified 95fcf680

Run cached graph api should grab device lock first (#4195)

JackCaoG committed 2 years ago

Verified 08a9c5dd

Add the dynamic shape unit test the CI. (#4180)

vanbasten23 committed 2 years ago

Verified a843d8bf

Older