Mojo module

device_context

Structs

• ​DeviceBuffer: Represents a block of device-resident storage. For GPU devices, a device buffer is allocated in the device's global memory.
• ​DeviceContext: Represents a single stream of execution on a particular accelerator (GPU). A DeviceContext serves as the low-level interface to the accelerator inside a MAX custom operation and provides methods for allocating buffers on the device, copying data between host and device, and for compiling and running functions (also known as kernels) on the device.
• ​DeviceExternalFunction: Represents an external device function loaded from PTX/SASS assembly.
• ​DeviceFunction: Represents a compiled device function for GPU execution.
• ​DeviceStream: Represents a CUDA/HIP stream for asynchronous GPU operations.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit