Add dropout to CMake

milancurcic · milancurcic · commit 544b23a2911c · 2025-01-23T17:11:18.000-05:00
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -51,6 +51,8 @@ add_library(neural-fortran
   src/nf/nf_reshape_layer_submodule.f90
   src/nf/io/nf_io_binary.f90
   src/nf/io/nf_io_binary_submodule.f90
+  src/nf/nf_dropout_layer.f90
+  src/nf/nf_dropout_layer_submodule.f90
 )
 
 target_link_libraries(neural-fortran PRIVATE)
diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
@@ -1,6 +1,7 @@
 foreach(execid
   input1d_layer
   input3d_layer
+  dropout_layer
   parametric_activation
   dense_layer
   conv2d_layer
diff --git a/test/test_dropout_layer.f90 b/test/test_dropout_layer.f90
@@ -87,7 +87,7 @@ program test_dropout_layer
       end if
     end do
     if (.not. ok) then
-      write(stderr, '(a)') 'dropout layer output sum should match input sum within 1% tolerance.. failed'
+      write(stderr, '(a)') 'dropout layer output sum should match input sum within tolerance.. failed'
     end if
   end block forward_pass
 

Original file line number	Diff line number	Diff line change
`@@ -51,6 +51,8 @@ add_library(neural-fortran`
`51`	`51`	`src/nf/nf_reshape_layer_submodule.f90`
`52`	`52`	`src/nf/io/nf_io_binary.f90`
`53`	`53`	`src/nf/io/nf_io_binary_submodule.f90`
	`54`	`+ src/nf/nf_dropout_layer.f90`
	`55`	`+ src/nf/nf_dropout_layer_submodule.f90`
`54`	`56`	`)`
`55`	`57`
`56`	`58`	`target_link_libraries(neural-fortran PRIVATE)`