mrope_get_position_ids / torch-ext /torch_binding.h
danieldk's picture
danieldk HF Staff
Sync with latest kernel-builder changes
b865e0f
Raw History Blame Contribute Delete
170 Bytes
#pragma once
#include <torch/torch.h>
void get_position_ids(torch::Tensor &out, torch::Tensor &input_ids,
torch::Tensor &image_grid_thw);