For the complete Mojo documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /docs/manual/basics.md).
is_nvidia_gpu
def is_nvidia_gpu[subarch: StringSpan[ImmStaticOrigin] = StringSpan("<any>")]() -> Bool
Checks whether the current compilation target is an NVIDIA GPU.
Equivalent to CompilationTarget.current().is_nvidia_gpu[subarch]().
Parameters:
- subarch (
StringSpan[ImmStaticOrigin]): The GPU sub-architecture to check for.
Returns:
Bool: True if the current compilation target is an NVIDIA GPU.