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
is_nvidia_gpu() -> Bool
Returns True if the target triple of the compiler is nvptx64-nvidia-cuda False otherwise.
Returns:
Bool: True if the triple target is cuda and False otherwise.
is_nvidia_gpu[subarch: StringSlice[StaticConstantOrigin]]() -> Bool
Returns True if the target triple of the compiler is nvptx64-nvidia-cuda and we are compiling for the specified sub-architecture and False otherwise.
Parameters:
- subarch (
StringSlice[StaticConstantOrigin]): The subarchitecture (e.g. sm_80).
Returns:
Bool: True if the triple target is cuda and False otherwise.