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
b1a9bab4
"test/vscode:/vscode.git/clone" did not exist on "a984872dfee71a553eb90c7eb13c9a5609d041c0"
Commit
b1a9bab4
authored
Feb 04, 2025
by
Aviral Goel
Browse files
added flex attention test for gfx90a and gfx942
parent
fbf2c582
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
22 additions
and
2 deletions
+22
-2
Jenkinsfile
Jenkinsfile
+22
-2
No files found.
Jenkinsfile
View file @
b1a9bab4
...
...
@@ -349,6 +349,7 @@ def cmake_build(Map conf=[:]){
echo
"could not locate the requested artifacts: ${err.getMessage()}. will skip the stashing."
}
}
// TODO: Do we do same for flex attention?
if
(
params
.
RUN_CK_TILE_GEMM_TESTS
){
try
{
archiveArtifacts
"perf_tile_gemm_*.log"
...
...
@@ -713,7 +714,7 @@ def process_results(Map conf=[:]){
}
//launch develop branch daily at 23:00 UT in FULL_QA mode and at 19:00 UT with latest staging compiler version
CRON_SETTINGS
=
BRANCH_NAME
==
"develop"
?
'''0 23 * * * % RUN_FULL_QA=true;ROCMVERSION=6.3;RUN_CK_TILE_FMHA_TESTS=true;;RUN_CK_TILE_GEMM_TESTS=true
CRON_SETTINGS
=
BRANCH_NAME
==
"develop"
?
'''0 23 * * * % RUN_FULL_QA=true;ROCMVERSION=6.3;RUN_CK_TILE_FMHA_TESTS=true;
RUN_CK_TILE_FLEX_ATTENTION_TESTS=true
;RUN_CK_TILE_GEMM_TESTS=true
0 21 * * * % ROCMVERSION=6.3;hipTensor_test=true;RUN_CODEGEN_TESTS=true
0 19 * * * % BUILD_DOCKER=true;DL_KERNELS=true;COMPILER_VERSION=amd-staging;BUILD_COMPILER=/llvm-project/build/bin/clang++;USE_SCCACHE=false;NINJA_BUILD_TRACE=true
0 17 * * * % BUILD_DOCKER=true;DL_KERNELS=true;COMPILER_VERSION=amd-mainline;BUILD_COMPILER=/llvm-project/build/bin/clang++;USE_SCCACHE=false;NINJA_BUILD_TRACE=true
...
...
@@ -1012,6 +1013,25 @@ pipeline {
cleanWs
()
}
}
stage
(
"Run RUN_CK_TILE_FLEX_ATTENTION_TESTS Tests on gfx942"
)
{
when
{
beforeAgent
true
expression
{
params
.
RUN_CK_TILE_FLEX_ATTENTION_TESTS
.
toBoolean
()
}
}
agent
{
label
rocmnode
(
"gfx942"
)
}
environment
{
setup_args
=
"NO_CK_BUILD"
execute_args
=
""" ../script/cmake-ck-dev.sh ../ gfx942 && \
make -j64 tile_example_fmha_fwd tile_example_fmha_bwd && \
cd ../ &&
eexample/ck_tile/18_flexattn/script/run_full_test.sh "CI_${params.COMPILER_VERSION}" "${env.BRANCH_NAME}" "${NODE_NAME}" gfx942 """
}
steps
{
buildHipClangJobAndReboot
(
setup_args:
setup_args
,
no_reboot:
true
,
build_type:
'Release'
,
execute_cmd:
execute_args
)
cleanWs
()
}
}
}
}
stage
(
"Run CK_TILE_GEMM Tests"
)
...
...
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