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")
# 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)
......
......@@ -5,17 +5,17 @@
#include "layernorm2d_fwd.hpp"
template <typename DataType,
ck_tile::index_t NRepeat,
ck_tile::index_t kNRepeat,
ck_tile::index_t kMThreadPerBlock,
ck_tile::index_t kNThreadPerBlock,
ck_tile::index_t VectorAccessSize,
ck_tile::index_t kVectorAccessSize,
bool kPadN,
bool kTwoPass = false>
using trait_ = layernorm2d_fwd_traits_<DataType,
NRepeat,
kNRepeat,
kMThreadPerBlock,
kNThreadPerBlock,
VectorAccessSize,
kVectorAccessSize,
kPadN,
false,
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