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 struct

TargetAccelerator

struct TargetAccelerator[gpu_info_: GPUInfo, target_: CompilationTarget, target_accelerator_values_: List[String] = List(__list_literal__=NoneType(None))]

Implemented traits​

AnyType, Defaultable, Deinitable, Movable, TargetAcceleratorType

comptime members​

compute​

comptime compute = gpu_info_.compute

default_accelerator​

comptime default_accelerator[Collections: TypeList[values]] = _LookupTargetAccelerator[comptime[C: TargetAcceleratorCollection, Elt: TargetAcceleratorType, Idx: Int] (C.normalize_target_arch(::StringSpan[ImmStaticOrigin, ::Origin[False, ImmStaticOrigin]()])(_accelerator_arch()) in Elt.target_accelerator_values), Collections].single_result

Lookup information for the default accelerator in the given target collections.

Parameters​

  • ​Collections (TypeList[values]):

from_arch​

comptime from_arch[name: StringSpan[ImmStaticOrigin], Collections: TypeList[values]] = _LookupTargetAccelerator[comptime[C: TargetAcceleratorCollection, Elt: TargetAcceleratorType, Idx: Int] (C.normalize_target_arch(::StringSpan[ImmStaticOrigin, ::Origin[False, ImmStaticOrigin]()])(name) in Elt.target_accelerator_values), Collections].single_result

Parameters​

  • ​name (StringSpan[ImmStaticOrigin]):
  • ​Collections (TypeList[values]):

from_gpu_name​

comptime from_gpu_name[name: StringSpan[ImmStaticOrigin], Collections: TypeList[values]] = _LookupTargetAccelerator[comptime[C: TargetAcceleratorCollection, Elt: TargetAcceleratorType, Idx: Int] (Elt.gpu_info.name == name), Collections].single_result

Parameters​

  • ​name (StringSpan[ImmStaticOrigin]):
  • ​Collections (TypeList[values]):

gpu_info​

comptime gpu_info = gpu_info_

mlir_target​

comptime mlir_target = _mlir_value

shared_memory_per_multiprocessor​

comptime shared_memory_per_multiprocessor = gpu_info_.shared_memory_per_multiprocessor

sm_count​

comptime sm_count = gpu_info_.sm_count

target​

comptime target = target_

target_accelerator_values​

comptime target_accelerator_values = target_accelerator_values_

Methods​

__init__​

def __init__(out self)

Was this page helpful?