Skip to content

Commit

Permalink
[OpenXLA] Add openxla.md (#5507)
Browse files Browse the repository at this point in the history
* Create openxla.md
  • Loading branch information
ManfeiBai authored Aug 30, 2023
1 parent 4582675 commit 02c4aff
Showing 1 changed file with 19 additions and 0 deletions.
19 changes: 19 additions & 0 deletions docs/openxla.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
# OpenXLA

As of June 28th, 2023, PyTorch/XLA now pulls XLA from OpenXLA.
OpenXLA is an [open source machine learning compiler XLA for GPUs, CPUs, and ML accelerators](https://github.com/openxla/xla).

Previous to OpenXLA, PyTorch/XLA pulled XLA directly from [TensorFlow](https://github.com/tensorflow/tensorflow/tree/master/tensorflow/compiler/xla). With our [XLA to OpenXLA migration](https://github.com/pytorch/xla/pull/5202), PyTorch/XLA now pulls XLA from [OpenXLA](https://github.com/openxla/xla).

# How to use OpenXLA

For [PJRT runtime](https://github.com/pytorch/xla/blob/master/docs/pjrt.md) users, there is no change with this migration. For XRT runtime users, there is a separate [XRT branch of PyTorch/XLA](https://github.com/pytorch/xla/tree/xrt) since OpenXLA doesn't support XRT.


# Performance
Below is a performance visual comparison of throughput for ResNet50 pre and post the migration on different TPU hardwares.

| | resnet50-pjrt-v2-8 | resnet50-pjrt-v4-8 | resnet50-pjrt-v4-32 |
| :------------ | :------------ | :------------ | :------------ |
| Pre Migration | 18.59 | 20.06 | 27.92 |
| Post Migration | 18.63 | 19.94 | 27.14 |

0 comments on commit 02c4aff

Please sign in to comment.