def dtype_to_cutlass_type(dtype: str):
    """Returns the corresponding cutlass type."""
    cutlass_type = DTYPE_TO_CUTLASSTYPE.get(dtype)

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