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

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

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