    def _size_to_vector_type(sz_in_byte) -> str:
        """return vector_type for the given size"""
        for vec_type, sz in vector_types[input_type]:
            if sz_in_byte % sz == 0:
                return vec_type
        raise NotImplementedError("Unsupported vector size: {}".format(sz_in_byte))
return vector_type for the given size
python/aitemplate/backend/cuda/reduce/reduce_small_axis.py