PyTorch: Compiling NumPy code into C++ or CUDA via torch.compile

pytorch.org

PyTorch: Compiling NumPy code into C++ or CUDA via torch.compile

pytorch.org

ylai@lemmy.ml to

Learn Machine Learning@sh.itjust.worksEnglish · 1 year ago

Compiling NumPy code into C++ or CUDA via torch.compile

pytorch.org

Quansight engineers have implemented support for tracing through NumPy code via torch.compile in PyTorch 2.1. This feature leverages PyTorch’s compiler to generate efficient fused vectorized code without having to modify your original NumPy code. Even more, it also allows for executing NumPy code on CUDA just by running it through torch.compile under torch.device(

You must log in or register to comment.

Chat

Learn Machine Learning@sh.itjust.works

learnmachinelearning@sh.itjust.works

Create a post

You are not logged in. However you can subscribe from another Fediverse account, for example Lemmy or Mastodon. To do this, paste the following into the search field of your instance: !learnmachinelearning@sh.itjust.works

Welcome! This is a place for people to learn more about machine learning techniques, discuss applications and ask questions.

Example questions:

“Should I use a deep neural network for my audio classification task?”
“I’m working with a small dataset, what can I do to make my model generalize well?”
“Is there a library available that implements function X in language Y?”
“I want to learn more about the math behind machine learning technique A, where should I start?”

Please do:

Be kind to new people
Post guides and tutorials that you find helpful
Link to open/free sources instead of paywalled when possible

Please don’t:

Post news articles / memes (there are other machine learning/AI communities for this)

Other communities in this area:

Similar subreddits: r/MLquestions, r/askmachinelearning, r/learnmachinelearning

Visibility: Public

This community can be federated to other instances and be posted/commented in by their users.

1 user / day
1 user / week
1 user / month
1 user / 6 months
1 local subscriber
522 subscribers
59 Posts
7 Comments
Modlog