def dtype_to_cuda_type(dtype: str):
    """Returns the corresponding cuda type."""
    cuda_type = DTYPE_TO_CUDATYPE.get(dtype)

    if cuda_type is None:
        raise NotImplementedError("CUDA - Unsupported dtype: {}".format(dtype))
    return cuda_type
Returns the corresponding cuda type.
python/aitemplate/backend/cuda/cuda_common.py