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
8c905605
Unverified
Commit
8c905605
authored
Jan 22, 2025
by
Max Podkorytov
Browse files
remove bwd related commands from cmakelists
parent
81e68c64
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
29 deletions
+2
-29
example/ck_tile/18_flexattn/CMakeLists.txt
example/ck_tile/18_flexattn/CMakeLists.txt
+2
-29
No files found.
example/ck_tile/18_flexattn/CMakeLists.txt
View file @
8c905605
...
...
@@ -28,19 +28,9 @@ if(ret AND NOT ret EQUAL 0)
message
(
FATAL_ERROR
"CK Tile FMHA FAILED to genrate a list of FWD kernels via Python."
)
endif
()
execute_process
(
COMMAND
${
Python3_EXECUTABLE
}
${
CMAKE_CURRENT_LIST_DIR
}
/generate.py
--api bwd --list_blobs
${
CMAKE_CURRENT_BINARY_DIR
}
/bwd_blob_list.txt --receipt 3
RESULT_VARIABLE ret
)
if
(
ret AND NOT ret EQUAL 0
)
message
(
FATAL_ERROR
"CK Tile FMHA FAILED to genrate a list of BWD kernels via Python."
)
endif
()
# NOTE: for cmake, the FMHA_FWD_GEN_BLOBS/FMHA_BWD_GEN_BLOBS files must be in the same directory
# NOTE: for cmake, the FMHA_FWD_GEN_BLOBS files must be in the same directory
# as current cmake list, otherwise will not figure out the dependency properly
file
(
STRINGS
${
CMAKE_CURRENT_BINARY_DIR
}
/fwd_blob_list.txt FMHA_FWD_GEN_BLOBS
)
file
(
STRINGS
${
CMAKE_CURRENT_BINARY_DIR
}
/bwd_blob_list.txt FMHA_BWD_GEN_BLOBS
)
add_custom_command
(
OUTPUT
${
FMHA_FWD_GEN_BLOBS
}
...
...
@@ -48,13 +38,7 @@ add_custom_command(
--api
${
FMHA_FWD_APIS
}
--output_dir
${
CMAKE_CURRENT_BINARY_DIR
}
)
add_custom_command
(
OUTPUT
${
FMHA_BWD_GEN_BLOBS
}
COMMAND
${
Python3_EXECUTABLE
}
${
CMAKE_CURRENT_LIST_DIR
}
/generate.py
--api bwd --output_dir
${
CMAKE_CURRENT_BINARY_DIR
}
--receipt 3
)
set
(
EXAMPLE_FMHA_FWD
"tile_example_fmha_fwd"
)
set
(
EXAMPLE_FMHA_FWD
"tile_example_flexattn_fwd"
)
# not using add_example_executable() to add this target, since we don't want this to have
# to be included in "make all/install/check"
message
(
"adding example
${
EXAMPLE_FMHA_FWD
}
"
)
...
...
@@ -62,13 +46,6 @@ add_executable(${EXAMPLE_FMHA_FWD} EXCLUDE_FROM_ALL fmha_fwd.cpp)
target_include_directories
(
${
EXAMPLE_FMHA_FWD
}
PRIVATE
${
CMAKE_CURRENT_LIST_DIR
}
)
target_sources
(
${
EXAMPLE_FMHA_FWD
}
PRIVATE
${
FMHA_FWD_GEN_BLOBS
}
)
set
(
EXAMPLE_FMHA_BWD
"tile_example_fmha_bwd"
)
# not using add_example_executable() to add this target, since we don't want this to have
# to be included in "make all/install/check"
message
(
"adding example
${
EXAMPLE_FMHA_BWD
}
"
)
add_executable
(
${
EXAMPLE_FMHA_BWD
}
EXCLUDE_FROM_ALL fmha_bwd.cpp
)
target_include_directories
(
${
EXAMPLE_FMHA_BWD
}
PRIVATE
${
CMAKE_CURRENT_LIST_DIR
}
)
target_sources
(
${
EXAMPLE_FMHA_BWD
}
PRIVATE
${
FMHA_BWD_GEN_BLOBS
}
)
# NOTE: this is dangerous since will change the whole kernel to flush denormals
# WIP with compiler team for an exp2 intrinsic..., then remove this
...
...
@@ -77,7 +54,6 @@ if(NOT DEFINED FMHA_FWD_FAST_EXP2)
endif
()
set
(
EXAMPLE_FMHA_FWD_COMPILE_OPTIONS
)
set
(
EXAMPLE_FMHA_BWD_COMPILE_OPTIONS
)
# NOTE: we turn off undefined-func-template to let source compile without explicit declare function specializations
# ... because they are auto-generated
...
...
@@ -86,7 +62,6 @@ if(FMHA_FWD_FAST_EXP2)
else
()
list
(
APPEND EXAMPLE_FMHA_FWD_COMPILE_OPTIONS -Wno-undefined-func-template -DCK_TILE_FMHA_FWD_FAST_EXP2=0
)
endif
()
list
(
APPEND EXAMPLE_FMHA_BWD_COMPILE_OPTIONS -Wno-undefined-func-template -fgpu-flush-denormals-to-zero
)
# conditionally enable call to the fwd_splitkv API in fmha_fwd example
if
(
"fwd_splitkv"
IN_LIST FMHA_FWD_ENABLE_APIS
)
...
...
@@ -104,10 +79,8 @@ endif()
# Allow comparing floating points directly in order to check sentinel values
list
(
APPEND EXAMPLE_FMHA_FWD_COMPILE_OPTIONS -Wno-float-equal
)
list
(
APPEND EXAMPLE_FMHA_BWD_COMPILE_OPTIONS -Wno-float-equal
)
target_compile_options
(
${
EXAMPLE_FMHA_FWD
}
PRIVATE
${
EXAMPLE_FMHA_FWD_COMPILE_OPTIONS
}
)
target_compile_options
(
${
EXAMPLE_FMHA_BWD
}
PRIVATE
${
EXAMPLE_FMHA_BWD_COMPILE_OPTIONS
}
)
# TODO: we have to turn off this global prop, otherwise the progress bar generated
# by cmake will print too many files, execvp: /bin/sh: Argument list too long
...
...
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