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

Introduce torch.compile to all PyTorch examples #2027

Open
champon1020 opened this issue Mar 12, 2024 · 0 comments
Open

Introduce torch.compile to all PyTorch examples #2027

champon1020 opened this issue Mar 12, 2024 · 0 comments

Comments

@champon1020
Copy link
Contributor

champon1020 commented Mar 12, 2024

As the previous discussion #2024 (comment), I would suggest to introduce torch.compile in PyTorch examples, following the migration of PyTorch to v2 (#2024).

I think it should be guaranteed to work torch.compile, which is the main feature of PyTorch v2 and able to speed up computation and memory access, on PyTorch operator.
However, as @tenzen-y mentioned at #2024 (comment), there is a concern about supported GPUs (or I think also CUDA versions).

Please let me know your opinions!

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

No branches or pull requests

1 participant