"vllm/multimodal/processing/processor.py" did not exist on "d0985c5feb212023fb61938cfc6b521843b6d031"
ops.h 254 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
#pragma once

#include <torch/csrc/stable/library.h>
#include <torch/csrc/stable/tensor.h>

#ifndef USE_ROCM
torch::stable::Tensor permute_cols(torch::stable::Tensor const& A,
                                   torch::stable::Tensor const& perm);
#endif