#pragma once

#include "operators/rope.h"
#include "utils/small_array.h"

namespace infini {

void rope_kernel(int dType, int *pos, void *input, void *output, int size,
                 int dim_model, int dim_head, int hidden_stride,
                 int pos_stride);

}; // namespace infini