Running code at GPU

How can I write a mojo code that is executed at the GPU instead of the CPU?
12 Replies
roboquant
roboquant12mo ago
That is currently not yet supported. They have been indicating that some insights how to do this will be revealed soon.
NickJJ
NickJJ12mo ago
Coming in two days!
No description
roboquant
roboquant12mo 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
NickJJ12mo ago
They will support NVIDIA hardware I guess, as they invited experts from NVIDIA to get talks. Good things are going to happen.
roboquant
roboquant12mo 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
NickJJ12mo ago
Stay tuned to the ModCon It's HERE!
NickJJ
NickJJ12mo ago
No description
James Usevitch
James Usevitch12mo 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
Xcur12mo ago
Chris Lattner
Chris Lattner12mo ago
We just "previewed" GPU support in Mojo, it still has a bit of development to go before being generally available
shelper
shelper12mo ago
is there a plan to support other GPUs? like GPUs on Android devices from QUALCOMM chipsets?
navhkrinn
navhkrinn12mo 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