Skip to content

Commit

Permalink
[ trivial ] clang format & revert some tmp codes
Browse files Browse the repository at this point in the history
- clang format is applied.
- revert Android.mk

Signed-off-by: Eunju Yang <[email protected]>
  • Loading branch information
EunjuYang committed Nov 20, 2024
1 parent da18596 commit e0bc0d8
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
7 changes: 3 additions & 4 deletions nntrainer/cl_context.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -44,9 +44,8 @@ static void add_default_object(ClContext &cc) {
// ml::train::LayerType::LAYER_ADDITION);

// @todo swiglulayercl also needs to be updated.
cc.registerFactory(nntrainer::createLayer<SwiGLULayerCl>,
SwiGLULayerCl::type,
ml::train::LayerType::LAYER_SWIGLU);
cc.registerFactory(nntrainer::createLayer<SwiGLULayerCl>, SwiGLULayerCl::type,
ml::train::LayerType::LAYER_SWIGLU);

if (ReshapeLayerCl::registerClKernels()) {
cc.registerFactory(nntrainer::createLayer<ReshapeLayerCl>,
Expand All @@ -56,7 +55,7 @@ static void add_default_object(ClContext &cc) {

// @todo rmsnormlayercl also needs to be updated.
cc.registerFactory(nntrainer::createLayer<RMSNormLayerCl>,
RMSNormLayerCl::type, ml::train::LayerType::LAYER_RMSNORM);
RMSNormLayerCl::type, ml::train::LayerType::LAYER_RMSNORM);

if (ConcatLayerCl::registerClKernels()) {
cc.registerFactory(nntrainer::createLayer<ConcatLayerCl>,
Expand Down
2 changes: 2 additions & 0 deletions test/jni/Android.mk
Original file line number Diff line number Diff line change
Expand Up @@ -444,11 +444,13 @@ LOCAL_SRC_FILES := \
../unittest/layers/unittest_layers_impl.cpp \
../unittest/layers/unittest_layers_transpose_cl.cpp \
../unittest/layers/unittest_layers_concat_cl.cpp \
../unittest/layers/unittest_layers_swiglu_cl.cpp \
../unittest/layers/unittest_layers_fully_connected_cl.cpp \
../unittest/layers/unittest_layers_input.cpp \
../unittest/layers/unittest_layers_loss.cpp \
../unittest/layers/unittest_layers_reshape_cl.cpp \
../unittest/layers/unittest_layers_fully_connected.cpp \
../unittest/layers/unittest_layers_rmsnorm_cl.cpp \
../unittest/layers/unittest_layers_batch_normalization.cpp \
../unittest/layers/unittest_layers_layer_normalization.cpp \
../unittest/layers/unittest_layers_convolution2d.cpp \
Expand Down

0 comments on commit e0bc0d8

Please sign in to comment.