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
af9546d9
Unverified
Commit
af9546d9
authored
Nov 09, 2024
by
Po Yen Chen
Committed by
GitHub
Nov 09, 2024
Browse files
Fix 'sh' command compatibility of smoke_test_fwd.sh (#1553)
parent
ea3640fd
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
10 deletions
+10
-10
example/ck_tile/01_fmha/script/smoke_test_fwd.sh
example/ck_tile/01_fmha/script/smoke_test_fwd.sh
+10
-10
No files found.
example/ck_tile/01_fmha/script/smoke_test_fwd.sh
View file @
af9546d9
...
@@ -29,14 +29,14 @@ while getopts ":sa" opt; do
...
@@ -29,14 +29,14 @@ while getopts ":sa" opt; do
done
done
run_fp16_bf16_tests
()
{
run_fp16_bf16_tests
()
{
local
NUM_SPLITS
=
(
1
)
local
NUM_SPLITS
=
"1"
local
PAGE_BLOCK_SIZE
=
(
0
)
local
PAGE_BLOCK_SIZE
=
"0"
local
CACHE_BATCH_IDX
=
(
0
)
local
CACHE_BATCH_IDX
=
"0"
if
[
$TEST_SPLITKV
-eq
1
]
;
then
if
[
$TEST_SPLITKV
-eq
1
]
;
then
NUM_SPLITS
+
=(
2 3
)
NUM_SPLITS
=
"
$NUM_SPLITS
2 3
"
PAGE_BLOCK_SIZE
+
=(
128
)
PAGE_BLOCK_SIZE
=
"
$PAGE_BLOCK_SIZE
128
"
CACHE_BATCH_IDX
+
=(
1
)
CACHE_BATCH_IDX
=
"
$CACHE_BATCH_IDX
1"
fi
fi
for
prec
in
"fp16"
"bf16"
;
do
for
prec
in
"fp16"
"bf16"
;
do
...
@@ -47,9 +47,9 @@ run_fp16_bf16_tests() {
...
@@ -47,9 +47,9 @@ run_fp16_bf16_tests() {
for
lse
in
0 1
;
do
for
lse
in
0 1
;
do
for
bias
in
"n"
"e"
"a"
;
do
for
bias
in
"n"
"e"
"a"
;
do
for
p_drop
in
0.0 0.2
;
do
for
p_drop
in
0.0 0.2
;
do
for
num_splits
in
"
${
NUM_SPLITS
[@]
}
"
;
do
for
num_splits
in
$
NUM_SPLITS
;
do
for
page_block_size
in
"
${
PAGE_BLOCK_SIZE
[@]
}
"
;
do
for
page_block_size
in
$
PAGE_BLOCK_SIZE
;
do
for
cache_batch_idx
in
"
${
CACHE_BATCH_IDX
[@]
}
"
;
do
for
cache_batch_idx
in
$
CACHE_BATCH_IDX
;
do
# $EXE -prec=$prec -mode=$mode -b=1 -h=1 -d=$hdim -s=1024 -bias=$bias -p_drop=$p_drop -lse=$lse -iperm=$perm -operm=$perm -vlayout=$vlayout -num_splits=$num_splits -page_block_size=$page_block_size -kname=$KNAME $COMMON_ARGS
# $EXE -prec=$prec -mode=$mode -b=1 -h=1 -d=$hdim -s=1024 -bias=$bias -p_drop=$p_drop -lse=$lse -iperm=$perm -operm=$perm -vlayout=$vlayout -num_splits=$num_splits -page_block_size=$page_block_size -kname=$KNAME $COMMON_ARGS
$EXE
-prec
=
$prec
-mode
=
$mode
-b
=
2
-h
=
2
-h_k
=
1
-d
=
16,
-d_v
=
$hdim
-s
=
55
-s_k
=
256
-bias
=
$bias
-p_drop
=
$p_drop
-lse
=
$lse
-iperm
=
$perm
-operm
=
$perm
-vlayout
=
$vlayout
-num_splits
=
$num_splits
-page_block_size
=
$page_block_size
-cache_batch_idx
=
$cache_batch_idx
-kname
=
$KNAME
$COMMON_ARGS
$EXE
-prec
=
$prec
-mode
=
$mode
-b
=
2
-h
=
2
-h_k
=
1
-d
=
16,
-d_v
=
$hdim
-s
=
55
-s_k
=
256
-bias
=
$bias
-p_drop
=
$p_drop
-lse
=
$lse
-iperm
=
$perm
-operm
=
$perm
-vlayout
=
$vlayout
-num_splits
=
$num_splits
-page_block_size
=
$page_block_size
-cache_batch_idx
=
$cache_batch_idx
-kname
=
$KNAME
$COMMON_ARGS
...
...
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