Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
ColossalAI
Commits
e014144c
Commit
e014144c
authored
Mar 31, 2022
by
Luxios22
Committed by
binmakeswell
Apr 06, 2022
Browse files
fix format (#565)
parent
1762ba14
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
6 deletions
+4
-6
colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h
...ai/kernel/cuda_native/csrc/kernels/include/feed_forward.h
+4
-6
No files found.
colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h
View file @
e014144c
...
@@ -13,16 +13,14 @@
...
@@ -13,16 +13,14 @@
#include "cublas_wrappers.h"
#include "cublas_wrappers.h"
#include "kernels.h"
#include "kernels.h"
template
<
typename
T
>
template
<
typename
T
>
class
FeedForward
{
class
FeedForward
{
public:
public:
struct
Config
{
struct
Config
{
int
outputSize
;
int
outputSize
;
int
inputSize
;
int
inputSize
;
std
::
array
<
int
,
3
>
gemm_algos
;
std
::
array
<
int
,
3
>
gemm_algos
;
Config
(
int
outputs
,
int
inputs
)
Config
(
int
outputs
,
int
inputs
)
:
outputSize
(
outputs
),
:
outputSize
(
outputs
),
inputSize
(
inputs
),
inputSize
(
inputs
),
gemm_algos
(
std
::
array
<
int
,
3
>
({
99
,
99
,
99
}))
{}
gemm_algos
(
std
::
array
<
int
,
3
>
({
99
,
99
,
99
}))
{}
};
};
...
@@ -63,6 +61,6 @@ class FeedForward {
...
@@ -63,6 +61,6 @@ class FeedForward {
config_
.
inputSize
=
inputSize
;
config_
.
inputSize
=
inputSize
;
}
}
private:
private:
Config
config_
;
Config
config_
;
};
};
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment