github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix format (#567)

pull/673/head
YuliangLiu0306 2022-03-31 15:05:58 +08:00 committed by binmakeswell
parent 0f1da44e5e
commit 9420d3ae31
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
colossalai/kernel/cuda_native/csrc/kernels/include/softmax.h
View File

@ -10,9 +10,8 @@
using namespace std; using namespace std;
template <typename T> template <typename T> class Softmax {
class Softmax { public:
public:
struct Config { struct Config {
size_t nhead; size_t nhead;
Config(size_t nhead) : nhead(nhead) {} Config(size_t nhead) : nhead(nhead) {}
@ -35,10 +34,8 @@ class Softmax {
stream); stream);
} }
void reset_size(size_t nhead) { void reset_size(size_t nhead) { config_.nhead = nhead; }
config_.nhead = nhead;
}
private: private:
Config config_; Config config_;
}; };