Running code at GPU

How can I write a mojo code that is executed at the GPU instead of the CPU?
12 Replies
roboquant
roboquant10mo ago
That is currently not yet supported. They have been indicating that some insights how to do this will be revealed soon.
NickJJ
NickJJ10mo ago
Coming in two days!
No description
roboquant
roboquant10mo ago
Curious what the extra hardware support entitles (if any). I get the feeling that Modular with their AI engine is first focusing on inference, so perhaps some Intel GPU/Habana hardware?
NickJJ
NickJJ10mo ago
They will support NVIDIA hardware I guess, as they invited experts from NVIDIA to get talks. Good things are going to happen.
roboquant
roboquant10mo ago
That would be great. If mojo gets indeed nvidia support on the short term, I can see machine learning in mojo becoming a thing quickly.
NickJJ
NickJJ10mo ago
Stay tuned to the ModCon It's HERE!
NickJJ
NickJJ10mo ago
No description
James Usevitch
James Usevitch10mo ago
I only caught part of the keynote, but it sounded to me as if Mojo GPU execution was only available through the MAX Engine. Is it possible to execute Mojo code on the GPU without MAX?
Xcur
Xcur10mo ago
Chris Lattner
Chris Lattner10mo ago
We just "previewed" GPU support in Mojo, it still has a bit of development to go before being generally available
shelper
shelper10mo ago
is there a plan to support other GPUs? like GPUs on Android devices from QUALCOMM chipsets?
navhkrinn
navhkrinn10mo ago
I'm just speculating but I remember they did say "stay tuned for GTC 2024" or something like that on ModCon. And it would be natural to announce GPU computing with all the fancy announcements that Nvidia will be making. That is of course just a speculation on my side
Want results from more Discord servers?
Add your server