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
fa54b344
Commit
fa54b344
authored
Feb 03, 2025
by
Aviral Goel
Browse files
added flex_attention in Jenkins file
parent
d480a5a6
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
29 additions
and
0 deletions
+29
-0
Jenkinsfile
Jenkinsfile
+29
-0
No files found.
Jenkinsfile
View file @
fa54b344
...
...
@@ -793,6 +793,10 @@ pipeline {
name:
"RUN_CK_TILE_FMHA_TESTS"
,
defaultValue:
false
,
description:
"Run the ck_tile FMHA tests (default: OFF)"
)
booleanParam
(
name:
"RUN_CK_TILE_FLEX_ATTENSION_TESTS"
,
defaultValue:
false
,
description:
"Run the ck_tile FLEX ATTENTION tests (default: OFF)"
)
booleanParam
(
name:
"RUN_CK_TILE_GEMM_TESTS"
,
defaultValue:
true
,
...
...
@@ -984,6 +988,31 @@ pipeline {
}
}
}
stage
(
"Run RUN_CK_TILE_FLEX_ATTENSION_TESTS Tests"
)
{
parallel
{
stage
(
"Run RUN_CK_TILE_FLEX_ATTENSION_TESTS Tests on gfx90a"
)
{
when
{
beforeAgent
true
expression
{
params
.
RUN_CK_TILE_FLEX_ATTENSION_TESTS
.
toBoolean
()
}
}
agent
{
label
rocmnode
(
"gfx90a"
)
}
environment
{
setup_args
=
"NO_CK_BUILD"
execute_args
=
""" ../script/cmake-ck-dev.sh ../ gfx90a && \
make -j64 tile_example_fmha_fwd && \
cd ../ &&
example/ck_tile/18_flexattn/script/run_full_test.sh "CI_${params.COMPILER_VERSION}" "${env.BRANCH_NAME}" "${NODE_NAME}" gfx90a """
}
steps
{
buildHipClangJobAndReboot
(
setup_args:
setup_args
,
no_reboot:
true
,
build_type:
'Release'
,
execute_cmd:
execute_args
)
cleanWs
()
}
}
}
}
stage
(
"Run CK_TILE_GEMM Tests"
)
{
parallel
...
...
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