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
22d1df22
Commit
22d1df22
authored
May 16, 2022
by
ExtremeViscent
Committed by
binmakeswell
May 17, 2022
Browse files
[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h (#968)
code style
parent
fb5bc6cb
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
4 deletions
+6
-4
colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h
...ai/kernel/cuda_native/csrc/kernels/include/feed_forward.h
+6
-4
No files found.
colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h
View file @
22d1df22
...
@@ -13,14 +13,16 @@
...
@@ -13,14 +13,16 @@
#include "cublas_wrappers.h"
#include "cublas_wrappers.h"
#include "kernels.h"
#include "kernels.h"
template
<
typename
T
>
class
FeedForward
{
template
<
typename
T
>
public:
class
FeedForward
{
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
),
inputSize
(
inputs
),
:
outputSize
(
outputs
),
inputSize
(
inputs
),
gemm_algos
(
std
::
array
<
int
,
3
>
({
99
,
99
,
99
}))
{}
gemm_algos
(
std
::
array
<
int
,
3
>
({
99
,
99
,
99
}))
{}
};
};
...
@@ -61,6 +63,6 @@ public:
...
@@ -61,6 +63,6 @@ public:
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