github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-07 20:39:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix format (#565)

Browse Source
This commit is contained in:
Luxios22 2022-03-31 15:00:21 +08:00 committed by binmakeswell
parent 1762ba14ab
commit e014144c44
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h
View File

@ -13,16 +13,14 @@
#include "cublas_wrappers.h"
#include "kernels.h"
template <typename T>
class FeedForward {
template <typename T> class FeedForward {
public:
struct Config {
int outputSize;
int inputSize;
std::array<int, 3> gemm_algos;
Config(int outputs, int inputs)
: outputSize(outputs),
inputSize(inputs),
: outputSize(outputs), inputSize(inputs),
gemm_algos(std::array<int, 3>({99, 99, 99})) {}
};