Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Pallas TPU] Add a note in the Pallas Quickstart documentation about the instructions of running the existing example on TPU #23889

Merged
merged 1 commit into from
Sep 27, 2024

Conversation

copybara-service[bot]
Copy link

@copybara-service copybara-service bot commented Sep 24, 2024

[Pallas TPU] Add a note in the Pallas Quickstart documentation about the instructions of running the existing example on TPU

This fixes #22817

This changes is originally proposed by @justinjfu in the comments of the above issue.

This PR is related to #23885.

@copybara-service copybara-service bot force-pushed the test_678420788 branch 9 times, most recently from d7bd3a3 to 1673ef9 Compare September 26, 2024 21:51
@copybara-service copybara-service bot changed the title [Pallas TPU] Add a note in the Pallas Quickstart documentation about the instructions of running the existing example on TPU and some explanations about the difference in memory handling between TPUs and GPUs [Pallas TPU] Add a note in the Pallas Quickstart documentation about the instructions of running the existing example on TPU Sep 26, 2024
@copybara-service copybara-service bot force-pushed the test_678420788 branch 3 times, most recently from abf2dd7 to 0028c19 Compare September 27, 2024 08:08
…the instructions of running the existing example on TPU

This fixes #22817

This changes is originally proposed by @justinjfu in the comments of the above issue.

This PR is related to #23885.

PiperOrigin-RevId: 679487218
@copybara-service copybara-service bot merged commit ab4590c into main Sep 27, 2024
1 of 2 checks passed
@copybara-service copybara-service bot deleted the test_678420788 branch September 27, 2024 08:33
copybara-service bot pushed a commit that referenced this pull request Oct 3, 2024
This PR uses the same method to add cross references as the previous PR #23889

---

The content below is for future references.

Useful commands:

```sh
# sync *.md to *.ipynb
jupytext docs/pallas/quickstart.md -o docs/pallas/quickstart.ipynb

# build documentation
sphinx-build -b html -D nb_execution_mode=off docs docs/build/html -j auto
```

Create labels in *.md:

```md
(pallas_block_specs_by_example)=
```

Reference a label in *.md

```md
{ref}`pallas_block_specs_by_example`
```

Create labels in *.rst:

```rst
.. _pallas_tpu_noteworthy_properties:
```

PiperOrigin-RevId: 681675621
copybara-service bot pushed a commit that referenced this pull request Oct 3, 2024
This PR uses the same method to add cross references as the previous PR #23889.

---

The content below is for future references.

Useful commands:

```sh
# sync *.md to *.ipynb
jupytext docs/pallas/quickstart.md -o docs/pallas/quickstart.ipynb

# build documentation
sphinx-build -b html -D nb_execution_mode=off docs docs/build/html -j auto
```

Create a label in *.md:

```md
(pallas_block_specs_by_example)=
```

Create a label in *.rst:

```rst
.. _pallas_tpu_noteworthy_properties:
```

Reference a label in *.md

```md
{ref}`pallas_block_specs_by_example`
```

PiperOrigin-RevId: 681675621
copybara-service bot pushed a commit that referenced this pull request Oct 3, 2024
This PR uses the same method to add cross references as the previous PR #23889.

---

The content below is for future references.

#### Useful commands

Build documentation:

```sh
sphinx-build -b html -D nb_execution_mode=off docs docs/build/html -j auto
```

Create a label in *.md:

```md
(pallas_block_specs_by_example)=
```

Create a label in *.rst:

```rst
.. _pallas_tpu_noteworthy_properties:
```

Reference a label in *.md

```md
{ref}`pallas_block_specs_by_example`
```

PiperOrigin-RevId: 681675621
copybara-service bot pushed a commit that referenced this pull request Oct 3, 2024
This PR uses the same method to add cross references as the previous PR #23889.

---

The content below is for future references.

#### Useful commands

Build documentation:

```sh
sphinx-build -b html -D nb_execution_mode=off docs docs/build/html -j auto
```

Create a label in *.md:

```md
(pallas_block_specs_by_example)=
```

Create a label in *.rst:

```rst
.. _pallas_tpu_noteworthy_properties:
```

Reference a label in *.md:

```md
{ref}`pallas_block_specs_by_example`
```

Sync changes from *.md to *.ipynb:

```sh
jupytext --sync docs/pallas/tpu/distributed.md
```

PiperOrigin-RevId: 681675621
copybara-service bot pushed a commit that referenced this pull request Oct 3, 2024
This PR uses the same method to add cross references as the previous PR #23889.

---

The content below is for future references.

#### Useful commands

Build documentation:

```sh
sphinx-build -b html -D nb_execution_mode=off docs docs/build/html -j auto
```

Create a label in *.md:

```md
(pallas_block_specs_by_example)=
```

Create a label in *.rst:

```rst
.. _pallas_tpu_noteworthy_properties:
```

Reference a label in *.md:

```md
{ref}`pallas_block_specs_by_example`
```

Sync changes from *.md to *.ipynb:

```sh
jupytext --sync docs/pallas/tpu/distributed.md
```

PiperOrigin-RevId: 681675621
copybara-service bot pushed a commit that referenced this pull request Oct 3, 2024
This PR uses the same method to add cross references as the previous PR #23889.

---

The content below is for future references.

#### Useful commands

Build documentation:

```sh
sphinx-build -b html -D nb_execution_mode=off docs docs/build/html -j auto
```

Create a label in *.md:

```md
(pallas_block_specs_by_example)=
```

Create a label in *.rst:

```rst
.. _pallas_tpu_noteworthy_properties:
```

Reference a label in *.md:

```md
{ref}`pallas_block_specs_by_example`
```

Sync changes from *.md to *.ipynb:

```sh
jupytext --sync docs/pallas/tpu/distributed.md
```

PiperOrigin-RevId: 682034607
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

pallas simple pl.program_id() example not working
1 participant