Skip to content

Commit 0165642

Browse files
committed
embedding_layer: update cmake
1 parent 10e54d0 commit 0165642

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

CMakeLists.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,8 @@ add_library(neural-fortran
4141
src/nf/nf_layer_submodule.f90
4242
src/nf/nf_linear2d_layer.f90
4343
src/nf/nf_linear2d_layer_submodule.f90
44+
src/nf/nf_embedding_layer.f90
45+
src/nf/nf_embedding_layer_submodule.f90
4446
src/nf/nf_loss.f90
4547
src/nf/nf_loss_submodule.f90
4648
src/nf/nf_maxpool2d_layer.f90

test/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ foreach(execid
1212
insert_flatten
1313
reshape_layer
1414
multihead_attention_layer
15+
embedding_layer
1516
dense_network
1617
get_set_network_params
1718
conv2d_network

0 commit comments

Comments
 (0)