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
1c870158
Commit
1c870158
authored
Oct 17, 2024
by
rocking
Browse files
Refine file name
parent
3b290001
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
6 additions
and
6 deletions
+6
-6
example/ck_tile/02_layernorm2d/CMakeLists.txt
example/ck_tile/02_layernorm2d/CMakeLists.txt
+2
-2
example/ck_tile/02_layernorm2d/instances/layernorm2d_fwd_api.cpp
.../ck_tile/02_layernorm2d/instances/layernorm2d_fwd_api.cpp
+4
-4
example/ck_tile/02_layernorm2d/layernorm2d_fwd.cpp
example/ck_tile/02_layernorm2d/layernorm2d_fwd.cpp
+0
-0
No files found.
example/ck_tile/02_layernorm2d/CMakeLists.txt
View file @
1c870158
...
...
@@ -3,9 +3,9 @@ set(EXAMPLE_LAYERNORM2D_FWD "tile_example_layernorm2d_fwd")
# to be included in "make all/install/check"
message
(
"adding example
${
EXAMPLE_LAYERNORM2D_FWD
}
"
)
file
(
GLOB INSTANCE_SRCS instances/*.cpp
)
add_executable
(
${
EXAMPLE_LAYERNORM2D_FWD
}
EXCLUDE_FROM_ALL
example_
layernorm2d_fwd.cpp
)
add_executable
(
${
EXAMPLE_LAYERNORM2D_FWD
}
EXCLUDE_FROM_ALL layernorm2d_fwd.cpp
)
target_include_directories
(
${
EXAMPLE_LAYERNORM2D_FWD
}
PRIVATE
${
CMAKE_CURRENT_LIST_DIR
}
)
target_sources
(
${
EXAMPLE_LAYERNORM2D_FWD
}
PRIVATE
layernorm2d_fwd_api.cpp
${
INSTANCE_SRCS
}
)
target_sources
(
${
EXAMPLE_LAYERNORM2D_FWD
}
PRIVATE
${
INSTANCE_SRCS
}
)
set
(
EXAMPLE_LAYERNORM2D_FWD_COMPILE_OPTIONS
)
...
...
example/ck_tile/02_layernorm2d/layernorm2d_fwd_api.cpp
→
example/ck_tile/02_layernorm2d/
instances/
layernorm2d_fwd_api.cpp
View file @
1c870158
...
...
@@ -5,17 +5,17 @@
#include "layernorm2d_fwd.hpp"
template
<
typename
DataType
,
ck_tile
::
index_t
NRepeat
,
ck_tile
::
index_t
k
NRepeat
,
ck_tile
::
index_t
kMThreadPerBlock
,
ck_tile
::
index_t
kNThreadPerBlock
,
ck_tile
::
index_t
VectorAccessSize
,
ck_tile
::
index_t
k
VectorAccessSize
,
bool
kPadN
,
bool
kTwoPass
=
false
>
using
trait_
=
layernorm2d_fwd_traits_
<
DataType
,
NRepeat
,
k
NRepeat
,
kMThreadPerBlock
,
kNThreadPerBlock
,
VectorAccessSize
,
k
VectorAccessSize
,
kPadN
,
false
,
kTwoPass
>
;
...
...
example/ck_tile/02_layernorm2d/
example_
layernorm2d_fwd.cpp
→
example/ck_tile/02_layernorm2d/layernorm2d_fwd.cpp
View file @
1c870158
File moved
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