Skip to content

Commit fc81884

Browse files
committed
Set up boilerplate
1 parent cd8ca0e commit fc81884

File tree

7 files changed

+11
-0
lines changed

7 files changed

+11
-0
lines changed

include/lbann/layers/misc/CMakeLists.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,5 +39,8 @@ set_full_path(THIS_DIR_HEADERS
3939
variance.hpp
4040
)
4141

42+
if (LBANN_HAS_DISTCONV)
43+
add_subdirectory(distconv)
44+
endif()
4245
# Propagate the files up the tree
4346
set(HEADERS "${HEADERS}" "${THIS_DIR_HEADERS}" PARENT_SCOPE)

include/lbann/layers/misc/distconv/CMakeLists.txt

Whitespace-only changes.

include/lbann/layers/misc/distconv/distonv_channelwise_softmax.hpp

Whitespace-only changes.

include/lbann/utils/distconv.hpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,12 @@
5454
#endif // DISTCONV_HAS_P2P
5555

5656
#include "lbann/layers/learning/distconv/distconv_layers.hpp"
57+
<<<<<<< HEAD
5758
#include "lbann/layers/math/distconv/distconv_matmul.hpp"
5859

60+
=======
61+
#include "lbann/layers/misc/distconv/distconv_channelwise_softmax.hpp"
62+
>>>>>>> f6433dba5 (Set up boilerplate)
5963
namespace lbann {
6064

6165
inline auto default_hydrogen_stream()

src/layers/misc/CMakeLists.txt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,10 @@ endif ()
6565
# Add the subdirectories
6666
add_subdirectory(cereal_registration)
6767

68+
if (LBANN_HAS_DISTCONV)
69+
add_subdirectory(distconv)
70+
endif()
71+
6872
# Propagate the files up the tree
6973
set(SOURCES "${SOURCES}" "${THIS_DIR_SOURCES}" PARENT_SCOPE)
7074
set(GPU_SOURCES "${GPU_SOURCES}" "${THIS_DIR_CU_SOURCES}" PARENT_SCOPE)

src/layers/misc/distconv/CMakeLists.txt

Whitespace-only changes.

src/layers/misc/distconv/distconv_channelwise_softmax.cpp

Whitespace-only changes.

0 commit comments

Comments
 (0)