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_ROCM
Commits
d5d7de90
Commit
d5d7de90
authored
Oct 21, 2024
by
rocking
Browse files
Use smaller N
parent
2bc76625
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
example/ck_tile/02_layernorm2d/script/smoke_test.sh
example/ck_tile/02_layernorm2d/script/smoke_test.sh
+2
-2
No files found.
example/ck_tile/02_layernorm2d/script/smoke_test.sh
View file @
d5d7de90
#!/bin/sh
# call from top of CK folder
EXE
=
./
build/
bin/tile_example_layernorm2d_fwd
EXE
=
./bin/tile_example_layernorm2d_fwd
for
pr_i
in
"fp16"
"bf16"
;
do
$EXE
-prec
=
$pr_i
-m
=
99
-n
=
13
...
...
@@ -26,5 +26,5 @@ $EXE -prec=$pr_i -m=7 -n=2734
$EXE
-prec
=
$pr_i
-m
=
1
-n
=
3182
$EXE
-prec
=
$pr_i
-m
=
9
-n
=
4096
$EXE
-prec
=
$pr_i
-m
=
3
-n
=
8192
$EXE
-prec
=
$pr_i
-m
=
1
-n
=
23
547
$EXE
-prec
=
$pr_i
-m
=
1
-n
=
10
547
done
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