def check_dtype(lhs, rhs):
    """Check if dtypes in the given workload are supported by CUTLASS."""
    # Only fp16 inputs are supported for now.
    return lhs.dtype == rhs.dtype and lhs.dtype == "float16" and rhs.dtype == "float16"