From 22d1df224d085517d66726351f6bb540958461bb Mon Sep 17 00:00:00 2001 From: ExtremeViscent Date: Mon, 16 May 2022 03:20:48 +0100 Subject: [PATCH] [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h (#968) code style --- .../cuda_native/csrc/kernels/include/feed_forward.h | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h b/colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h index 9a43aeec3..ec963259f 100644 --- a/colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h +++ b/colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h @@ -13,14 +13,16 @@ #include "cublas_wrappers.h" #include "kernels.h" -template class FeedForward { -public: +template +class FeedForward { + public: struct Config { int outputSize; int inputSize; std::array gemm_algos; Config(int outputs, int inputs) - : outputSize(outputs), inputSize(inputs), + : outputSize(outputs), + inputSize(inputs), gemm_algos(std::array({99, 99, 99})) {} }; @@ -61,6 +63,6 @@ public: config_.inputSize = inputSize; } -private: + private: Config config_; };