Mojo function
get_cudnn_dtype
get_cudnn_dtype[dtype: DType]() -> cudnnDataType_t
Map Mojo DType to cuDNN data type.
Support only floating point dtypes for now.
Returns:
cudnnDataType_t
Raises:
If the dtype is not supported by cuDNN.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!