mirror of https://github.com/hpcaitech/ColossalAI
[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/softmax.h code style (#964)
parent
5bd80b7dd1
commit
7106a399fc
|
@ -10,7 +10,8 @@
|
|||
|
||||
using namespace std;
|
||||
|
||||
template <typename T> class Softmax {
|
||||
template <typename T>
|
||||
class Softmax {
|
||||
public:
|
||||
struct Config {
|
||||
size_t nhead;
|
||||
|
|
Loading…
Reference in New Issue