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.

Packages​

  • ​compute: GPU compute operations package - MMA and tensor core operations.
  • ​host: Implements the gpu host package.
  • ​memory: GPU memory operations package.
  • ​primitives: GPU primitives package - block, cluster, and grid-level operations.
  • ​sync: GPU synchronization primitives package.