IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /get-started.md). For the complete documentation index, see llms.txt.
Skip to main content
For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo package

gpu

Provides GPU programming primitives for hardware-accelerated Mojo code.

This package is moving from the Mojo standard library. Only the subpackages listed below have moved so far; the remaining gpu APIs are still documented in the Mojo standard library reference.

Packagesโ€‹

  • โ€‹compute: GPU compute operations package - MMA and tensor core operations.
  • โ€‹host: Implements the gpu host package.