Mojo module

device_context

This module provides functionality for interacting with accelerators. In particular the DeviceContext struct, which represents a single stream of execution on a given accelerator. You can use this struct to allocate accelerator memory, copy data to and from the accelerator, and compile and execute functions on the accelerator.

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).
• ​DeviceExternalFunction: Represents an external device function loaded from PTX/SASS assembly.
• ​DeviceFunction: Represents a compiled device function for GPU execution.
• ​DeviceMulticastBuffer: Represents a muticast memory object enables special memory operations to be broadcast across a group of devices.
• ​DeviceStream: Represents a CUDA/HIP stream for asynchronous GPU operations.
• ​HostBuffer: Represents a block of host-resident storage. For GPU devices, a host buffer is allocated in the host's global memory.

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