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
gaoqiong
composable_kernel
Commits
a22296fd
Commit
a22296fd
authored
Apr 07, 2023
by
rocking
Browse files
Refine problem size in example
parent
af2004b5
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
10 additions
and
10 deletions
+10
-10
client_example/18_groupnorm/groupnorm_swish.cpp
client_example/18_groupnorm/groupnorm_swish.cpp
+5
-5
example/42_groupnorm/run_groupnorm_example.inc
example/42_groupnorm/run_groupnorm_example.inc
+5
-5
No files found.
client_example/18_groupnorm/groupnorm_swish.cpp
View file @
a22296fd
...
@@ -40,11 +40,11 @@ struct SimpleDeviceMem
...
@@ -40,11 +40,11 @@ struct SimpleDeviceMem
int
main
(
int
argc
,
char
*
argv
[])
int
main
(
int
argc
,
char
*
argv
[])
{
{
ck
::
index_t
N
=
2
;
ck
::
index_t
N
=
3
2
;
ck
::
index_t
H
=
32
;
ck
::
index_t
H
=
16
;
ck
::
index_t
W
=
32
;
ck
::
index_t
W
=
16
;
ck
::
index_t
G
=
32
;
ck
::
index_t
G
=
64
;
ck
::
index_t
C
=
30
;
ck
::
index_t
C
=
128
;
std
::
size_t
xy_size
=
N
*
H
*
W
*
G
*
C
;
std
::
size_t
xy_size
=
N
*
H
*
W
*
G
*
C
;
std
::
size_t
gamma_beta_size
=
G
*
C
;
std
::
size_t
gamma_beta_size
=
G
*
C
;
...
...
example/42_groupnorm/run_groupnorm_example.inc
View file @
a22296fd
...
@@ -5,11 +5,11 @@
...
@@ -5,11 +5,11 @@
int
run_groupnorm_example
(
int
argc
,
char
*
argv
[])
int
run_groupnorm_example
(
int
argc
,
char
*
argv
[])
{
{
ck
::
index_t
N
=
2
;
ck
::
index_t
N
=
3
2
;
ck
::
index_t
H
=
32
;
ck
::
index_t
H
=
16
;
ck
::
index_t
W
=
32
;
ck
::
index_t
W
=
16
;
ck
::
index_t
G
=
32
;
ck
::
index_t
G
=
64
;
ck
::
index_t
C
=
30
;
ck
::
index_t
C
=
128
;
if
(
argc
==
1
)
if
(
argc
==
1
)
{
{
...
...
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