Skip to main content

Mojo package

memory

GPU memory operations package.

This package provides GPU memory operations including:

  • async_copy: Asynchronous memory copy operations
  • cp_async_bulk: Bulk tensor memory copy operations (TMA)
  • load/store: Memory access operations with cache control
  • fence operations: Memory ordering and synchronization
  • mbarrier: Memory barrier operations

These operations enable efficient data movement between different GPU memory spaces and coordination of memory operations across threads.

Modules

  • memory: This module provides GPU memory operations and utilities.

Was this page helpful?