10 lines
254 B
C
10 lines
254 B
C
|
#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
|