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

[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/softmax.h code style (#964)

pull/997/head
Yuer867 2022-05-15 18:06:08 +08:00 committed by binmakeswell
parent 5bd80b7dd1
commit 7106a399fc
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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