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 function

get_gpu_info

def get_gpu_info[target_arch: StringSpan[ImmStaticOrigin] = _accelerator_arch()]() -> GPUInfo

Gets GPUInfo for a specific target architecture.

Maps target architecture strings to corresponding GPUInfo instances.

Parameters:

• ​target_arch (StringSpan[ImmStaticOrigin]): Target architecture name (e.g., "sm_80", "gfx942"). Defaults to the current accelerator architecture.

Returns:

GPUInfo: GPUInfo instance for the specified target architecture.