Skip to content
This repository was archived by the owner on Aug 7, 2024. It is now read-only.

add wait_tensor() after all_gather in float8 to fix mem leak #261

Closed
wants to merge 2 commits into from

Conversation

bdhirsh
Copy link
Contributor

@bdhirsh bdhirsh commented May 16, 2024

I'm going to write a more detailed post internally to explain this memory leak

vkuzo and others added 2 commits May 15, 2024 12:34
…tp + sp

Summary:

Test Plan:

```
./test/test_dtensor.sh
```

Reviewers:

Subscribers:

Tasks:

Tags:
@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label May 16, 2024
@bdhirsh bdhirsh closed this May 16, 2024
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants