github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-31 19:35:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix format (#611)

Browse Source
This commit is contained in:
Yuer867 2022-04-01 14:19:27 +08:00 committed by binmakeswell
parent d3d5bedc65
commit 5ecef13c16
2 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
colossalai/kernel/cuda_native/csrc/kernels/include/cross_entropy_layer.h
View File

@ -8,8 +8,7 @@
#include "cuda_util.h"
template <typename T>
class CrossEntropyLayer {
template <typename T> class CrossEntropyLayer {
public:
CrossEntropyLayer(float epsilon, int padding_idx, int max_batch_tokens);