diff --git a/aten/src/ATen/native/metal/MetalPrepackOpContext.h b/aten/src/ATen/native/metal/MetalPrepackOpContext.h index a009e888bf43..04c6da7a3aed 100644 --- a/aten/src/ATen/native/metal/MetalPrepackOpContext.h +++ b/aten/src/ATen/native/metal/MetalPrepackOpContext.h @@ -70,7 +70,7 @@ class Conv2dOpContext : public torch::jit::CustomClassHolder { // The MPSCNNConvolution class takes weights in the order // [outputChannels][kernelHeight][kernelWidth][inputChannels/groups]. -static std::vector permuteWeights( +static inline std::vector permuteWeights( const float* src, const std::vector& sizes) { const int64_t M = sizes[0];