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

fix format (#567)

Browse Source
This commit is contained in:
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; 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;
@ -35,9 +34,7 @@ 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_;