Skip to content

Commit b3cdf0c

Browse files
committed
fix cmake, default with err
1 parent 00d79bb commit b3cdf0c

6 files changed

Lines changed: 6 additions & 9 deletions

File tree

CMakeLists.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,5 +51,4 @@ endforeach()
5151
add_subdirectory(app)
5252
add_subdirectory(include)
5353
add_subdirectory(src)
54-
add_subdirectory(src/layers_oneDNN)
5554
add_subdirectory(test)

src/CMakeLists.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,5 @@ add_subdirectory(graph)
22
add_subdirectory(graph_transformations)
33
add_subdirectory(perf)
44
add_subdirectory(layers)
5-
add_subdirectory(Weights_Reader)
5+
add_subdirectory(layers_oneDNN)
6+
add_subdirectory(Weights_Reader)

src/layers/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
file(GLOB_RECURSE layers_src *.cpp)
22
add_library(layers_lib STATIC "${LAYERS_HEADERS}" "${layers_src}")
33
target_link_libraries(layers_lib PUBLIC TBB_unified)
4-
target_link_libraries(layers_lib PUBLIC dnnl)
4+
target_link_libraries(layers_lib PUBLIC dnnl)

src/layers_oneDNN/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,4 +6,4 @@ target_include_directories(layers_oneDNN_lib PUBLIC
66
)
77
target_compile_definitions(layers_oneDNN_lib PRIVATE
88
DNNL_ENABLE_CONCURRENT_EXEC=1
9-
)
9+
)

src/layers_oneDNN/EWLayer_oneDNN.cpp

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -73,8 +73,8 @@ void EWLayer_oneDNN::initialize_onednn(const Shape& shape) {
7373
format = dnnl::memory::format_tag::abcde;
7474
break;
7575
default:
76-
format = dnnl::memory::format_tag::a;
77-
break;
76+
throw std::invalid_argument("Unsupported tensor dimensionality: " +
77+
std::to_string(dims.size()));
7878
}
7979

8080
memory_desc_ =
@@ -100,8 +100,6 @@ void EWLayer_oneDNN::initialize_onednn(const Shape& shape) {
100100

101101
initialized_ = true;
102102

103-
std::cout << "oneDNN EWLayer initialized for function: " << func_
104-
<< " with shape: [";
105103
for (size_t i = 0; i < dims.size(); ++i) {
106104
std::cout << dims[i];
107105
if (i < dims.size() - 1) std::cout << ", ";

test/CMakeLists.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,4 +55,3 @@ target_compile_definitions(run_test PRIVATE TEST_DATA_PATH="${CMAKE_SOURCE_DIR}/
5555

5656

5757
target_compile_definitions(run_test PUBLIC TESTS_BINARY_PATH="${CMAKE_CURRENT_BINARY_DIR}")
58-

0 commit comments

Comments
 (0)