pytorch/xla

Pull Requests Commits

Update artifacts.auto.tfvars for xrt

ManfeiBai committed 2 years ago

Verified 1a5fea53

Update artifacts.auto.tfvars night back

ManfeiBai committed 2 years ago

Verified ec48cbe1

Update artifacts.auto.tfvars

ManfeiBai committed 2 years ago

Verified ff6e4f67

Delete unimplemented methods from `ComputationClient` (#5639)

will-cromar committed 2 years ago

Verified ea5073b4

CI hotfix - Do not parallelize coverage runs (#5648)

yeounoh committed 2 years ago

Verified 1b68776e

Disable AI on master branch. (#5650)

vanbasten23 committed 2 years ago

Verified a6365dc2

Revert "Disable ABI on the master branch. (#5641)" (#5649)

vanbasten23 committed 2 years ago

Verified a1e76b45

lower native_dropout (#5643)

zpcore committed 2 years ago

Verified 2c6e4a77

Remove legacy StreamExecutor TPU client from the build (#5640)

will-cromar committed 2 years ago

Verified 00272c34

Disable ABI on the master branch. (#5641)

vanbasten23 committed 2 years ago

Verified a0a3aba9

lower NativeDropoutBackward (#5642)

JackCaoG committed 2 years ago

Verified d4966dbb

Parallelize CI test runs (#5626)

yeounoh committed 2 years ago

Verified 8412ca7f

Adding link to API_Guide and renaming the doc (#5632)

ssusie committed 2 years ago

Verified b5b902f7

Publish dynamic shape doc. (#5631)

vanbasten23 committed 2 years ago

Verified 5a73f9e3

[Docs] Fix small typo in README about DDP (#5637)

rohan-varma committed 2 years ago

Verified 987873c4

Stabilize Constant hash across processes (#5633)

jonb377 committed 2 years ago

Verified 9e14eaa6

Update TPU PCI vendor IDs (#5634)

will-cromar committed 2 years ago

Verified 65cd101f

Fix accidental override of XRT PyTorch branch (#5629)

will-cromar committed 2 years ago

Verified e14bdc80

Create cuda 12.1 dev trigger. (#5628)

vanbasten23 committed 2 years ago

Verified 2fda248f

Adding tutorial doc that should help users to get started and understand how code change, perf analysis and optimization can be done. (#5613)

ssusie committed 2 years ago

Verified 9f0d625e

Update the amp.md according to the latest changes (#5627)

yeounoh committed 2 years ago

Verified ac8e2bb6

Flatten `Computation` classes (#5230)

will-cromar committed 2 years ago

Verified caa96de7

Cache OpSharding by mesh and partition_spec (#5593)

jonb377 committed 2 years ago

Verified 71999f59

autocat test helper dtype default needs to be unset (#5622)

yeounoh committed 2 years ago

Verified 63a90204

Fix log spam when libtpu is loaded (#5619)

will-cromar committed 2 years ago

Verified eee1f2dd

Add cuda==12.1 trigger (#5617)

vanbasten23 committed 2 years ago

Verified 45bdf70b

Remove `on_device_shape` and `logical_on_device_shape` (#5546)

will-cromar committed 2 years ago

Verified dbb92c96

Clean up the redundant arg_min/max releated file (#5605)

zpcore committed 2 years ago

Verified a4874e2d

Allow running with bfloat16 on XLA:GPU autocast (#5598)

yeounoh committed 2 years ago

Verified b214c104

increase the GPU coverge timeout to be same as the GPU test (#5610)

JackCaoG committed 2 years ago

Verified e9c38170

Older