Mojo function
is_nvidia_gpu
is_nvidia_gpu() -> Bool
Returns True if the target triple of the compiler is nvptx64-nvidia-cuda False otherwise.
Returns:
True if the triple target is cuda and False otherwise.
is_nvidia_gpu[subarch: StringLiteral]() -> 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 (
StringLiteral): The subarchitecture (e.g. sm_80).
Returns:
True if the triple target is cuda and False otherwise.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!