Commit 1c870158 authored by rocking's avatar rocking
Browse files

Refine file name

parent 3b290001
...@@ -3,9 +3,9 @@ set(EXAMPLE_LAYERNORM2D_FWD "tile_example_layernorm2d_fwd") ...@@ -3,9 +3,9 @@ set(EXAMPLE_LAYERNORM2D_FWD "tile_example_layernorm2d_fwd")
# to be included in "make all/install/check" # to be included in "make all/install/check"
message("adding example ${EXAMPLE_LAYERNORM2D_FWD}") message("adding example ${EXAMPLE_LAYERNORM2D_FWD}")
file(GLOB INSTANCE_SRCS instances/*.cpp) 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_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) set(EXAMPLE_LAYERNORM2D_FWD_COMPILE_OPTIONS)
......
...@@ -5,17 +5,17 @@ ...@@ -5,17 +5,17 @@
#include "layernorm2d_fwd.hpp" #include "layernorm2d_fwd.hpp"
template <typename DataType, template <typename DataType,
ck_tile::index_t NRepeat, ck_tile::index_t kNRepeat,
ck_tile::index_t kMThreadPerBlock, ck_tile::index_t kMThreadPerBlock,
ck_tile::index_t kNThreadPerBlock, ck_tile::index_t kNThreadPerBlock,
ck_tile::index_t VectorAccessSize, ck_tile::index_t kVectorAccessSize,
bool kPadN, bool kPadN,
bool kTwoPass = false> bool kTwoPass = false>
using trait_ = layernorm2d_fwd_traits_<DataType, using trait_ = layernorm2d_fwd_traits_<DataType,
NRepeat, kNRepeat,
kMThreadPerBlock, kMThreadPerBlock,
kNThreadPerBlock, kNThreadPerBlock,
VectorAccessSize, kVectorAccessSize,
kPadN, kPadN,
false, false,
kTwoPass>; kTwoPass>;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment