/*************************************************************************
 * Copyright (C) [2023-2024] by Cambricon, Inc.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
 * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 *************************************************************************/
#ifndef CSRC_TORCH_API_GLUE_OPS_H_
#define CSRC_TORCH_API_GLUE_OPS_H_
#include "torch/extension.h"
#include "utils.h"

void copy_blocks__functionalization_glue(
    const std::vector<torch::Tensor> &k_caches,
    const std::vector<torch::Tensor> &v_caches,
    const c10::Dict<int64_t, c10::List<int64_t>> &block_mapping);

#endif  // CSRC_TORCH_API_GLUE_OPS_H_
