Commit b0421577 authored by Fenglei's avatar Fenglei Committed by Scott Cyphers

add cudnn include path to compiler.cpp (#902)

* add cudnn include path to compiler.cpp

* seperate two path

* Skipping one_hot tests for CPU as
CI is failing. JIRA bug report: https://jira01.devtools.intel.com/browse/NGRAPH-1682.
parent da1609b9
...@@ -287,6 +287,8 @@ endif() ...@@ -287,6 +287,8 @@ endif()
) )
set_property(SOURCE codegen/compiler.cpp APPEND_STRING PROPERTY COMPILE_DEFINITIONS set_property(SOURCE codegen/compiler.cpp APPEND_STRING PROPERTY COMPILE_DEFINITIONS
"CUDA_HEADER_PATHS=\"${CUDA_INCLUDE_DIRS}\";") "CUDA_HEADER_PATHS=\"${CUDA_INCLUDE_DIRS}\";")
set_property(SOURCE codegen/compiler.cpp APPEND_STRING PROPERTY COMPILE_DEFINITIONS
"CUDNN_HEADER_PATHS=\"${CUDNN_INCLUDE_DIRS}\";")
endif() endif()
endif() endif()
......
...@@ -419,6 +419,11 @@ void codegen::StaticCompiler::configure_search_path() ...@@ -419,6 +419,11 @@ void codegen::StaticCompiler::configure_search_path()
add_header_search_path(CUDA_HEADER_PATHS); add_header_search_path(CUDA_HEADER_PATHS);
#endif #endif
#ifdef CUDNN_HEADER_PATHS
// Only needed for GPU backend
add_header_search_path(CUDNN_HEADER_PATHS);
#endif
#ifdef NGRAPH_DISTRIBUTED #ifdef NGRAPH_DISTRIBUTED
add_header_search_path(MPI_HEADER_PATH); add_header_search_path(MPI_HEADER_PATH);
#endif #endif
......
...@@ -3384,6 +3384,7 @@ TEST(${BACKEND_NAME}, replace_slice_vector) ...@@ -3384,6 +3384,7 @@ TEST(${BACKEND_NAME}, replace_slice_vector)
TEST(${BACKEND_NAME}, one_hot_scalar_2_in_3) TEST(${BACKEND_NAME}, one_hot_scalar_2_in_3)
{ {
SKIP_TEST_FOR("CPU", "${BACKEND_NAME}");
Shape shape_a{}; Shape shape_a{};
auto A = make_shared<op::Parameter>(element::i32, shape_a); auto A = make_shared<op::Parameter>(element::i32, shape_a);
Shape shape_r{3}; Shape shape_r{3};
...@@ -3403,6 +3404,7 @@ TEST(${BACKEND_NAME}, one_hot_scalar_2_in_3) ...@@ -3403,6 +3404,7 @@ TEST(${BACKEND_NAME}, one_hot_scalar_2_in_3)
TEST(${BACKEND_NAME}, one_hot_scalar_1_in_3) TEST(${BACKEND_NAME}, one_hot_scalar_1_in_3)
{ {
SKIP_TEST_FOR("CPU", "${BACKEND_NAME}");
Shape shape_a{}; Shape shape_a{};
auto A = make_shared<op::Parameter>(element::i32, shape_a); auto A = make_shared<op::Parameter>(element::i32, shape_a);
Shape shape_r{3}; Shape shape_r{3};
...@@ -3422,6 +3424,7 @@ TEST(${BACKEND_NAME}, one_hot_scalar_1_in_3) ...@@ -3422,6 +3424,7 @@ TEST(${BACKEND_NAME}, one_hot_scalar_1_in_3)
TEST(${BACKEND_NAME}, one_hot_scalar_0_in_3) TEST(${BACKEND_NAME}, one_hot_scalar_0_in_3)
{ {
SKIP_TEST_FOR("CPU", "${BACKEND_NAME}");
Shape shape_a{}; Shape shape_a{};
auto A = make_shared<op::Parameter>(element::i32, shape_a); auto A = make_shared<op::Parameter>(element::i32, shape_a);
Shape shape_r{3}; Shape shape_r{3};
...@@ -3442,6 +3445,7 @@ TEST(${BACKEND_NAME}, one_hot_scalar_0_in_3) ...@@ -3442,6 +3445,7 @@ TEST(${BACKEND_NAME}, one_hot_scalar_0_in_3)
TEST(${BACKEND_NAME}, one_hot_scalar_fp_nonint_in_3) TEST(${BACKEND_NAME}, one_hot_scalar_fp_nonint_in_3)
{ {
SKIP_TEST_FOR("GPU", "${BACKEND_NAME}"); SKIP_TEST_FOR("GPU", "${BACKEND_NAME}");
SKIP_TEST_FOR("CPU", "${BACKEND_NAME}");
Shape shape_a{}; Shape shape_a{};
auto A = make_shared<op::Parameter>(element::f32, shape_a); auto A = make_shared<op::Parameter>(element::f32, shape_a);
Shape shape_r{3}; Shape shape_r{3};
...@@ -3471,6 +3475,7 @@ TEST(${BACKEND_NAME}, one_hot_scalar_fp_nonint_in_3) ...@@ -3471,6 +3475,7 @@ TEST(${BACKEND_NAME}, one_hot_scalar_fp_nonint_in_3)
TEST(${BACKEND_NAME}, one_hot_scalar_oob_in_3) TEST(${BACKEND_NAME}, one_hot_scalar_oob_in_3)
{ {
SKIP_TEST_FOR("CPU", "${BACKEND_NAME}");
SKIP_TEST_FOR("GPU", "${BACKEND_NAME}"); SKIP_TEST_FOR("GPU", "${BACKEND_NAME}");
SKIP_TEST_FOR("NNP_TESTER", "${BACKEND_NAME}"); SKIP_TEST_FOR("NNP_TESTER", "${BACKEND_NAME}");
...@@ -3503,6 +3508,7 @@ TEST(${BACKEND_NAME}, one_hot_scalar_oob_in_3) ...@@ -3503,6 +3508,7 @@ TEST(${BACKEND_NAME}, one_hot_scalar_oob_in_3)
TEST(${BACKEND_NAME}, one_hot_vector_0) TEST(${BACKEND_NAME}, one_hot_vector_0)
{ {
SKIP_TEST_FOR("CPU", "${BACKEND_NAME}");
Shape shape_a{8}; Shape shape_a{8};
auto A = make_shared<op::Parameter>(element::i32, shape_a); auto A = make_shared<op::Parameter>(element::i32, shape_a);
Shape shape_r{3, 8}; Shape shape_r{3, 8};
...@@ -3524,6 +3530,7 @@ TEST(${BACKEND_NAME}, one_hot_vector_0) ...@@ -3524,6 +3530,7 @@ TEST(${BACKEND_NAME}, one_hot_vector_0)
TEST(${BACKEND_NAME}, one_hot_vector_1) TEST(${BACKEND_NAME}, one_hot_vector_1)
{ {
SKIP_TEST_FOR("CPU", "${BACKEND_NAME}");
Shape shape_a{8}; Shape shape_a{8};
auto A = make_shared<op::Parameter>(element::i32, shape_a); auto A = make_shared<op::Parameter>(element::i32, shape_a);
Shape shape_r{8, 3}; Shape shape_r{8, 3};
...@@ -3545,6 +3552,7 @@ TEST(${BACKEND_NAME}, one_hot_vector_1) ...@@ -3545,6 +3552,7 @@ TEST(${BACKEND_NAME}, one_hot_vector_1)
TEST(${BACKEND_NAME}, one_hot_vector_1_barely_oob) TEST(${BACKEND_NAME}, one_hot_vector_1_barely_oob)
{ {
SKIP_TEST_FOR("CPU", "${BACKEND_NAME}");
SKIP_TEST_FOR("GPU", "${BACKEND_NAME}"); SKIP_TEST_FOR("GPU", "${BACKEND_NAME}");
Shape shape_a{8}; Shape shape_a{8};
auto A = make_shared<op::Parameter>(element::i32, shape_a); auto A = make_shared<op::Parameter>(element::i32, shape_a);
...@@ -3575,6 +3583,7 @@ TEST(${BACKEND_NAME}, one_hot_vector_1_barely_oob) ...@@ -3575,6 +3583,7 @@ TEST(${BACKEND_NAME}, one_hot_vector_1_barely_oob)
TEST(${BACKEND_NAME}, one_hot_vector_1_far_oob) TEST(${BACKEND_NAME}, one_hot_vector_1_far_oob)
{ {
SKIP_TEST_FOR("CPU", "${BACKEND_NAME}");
SKIP_TEST_FOR("GPU", "${BACKEND_NAME}"); SKIP_TEST_FOR("GPU", "${BACKEND_NAME}");
SKIP_TEST_FOR("NNP_TESTER", "${BACKEND_NAME}"); SKIP_TEST_FOR("NNP_TESTER", "${BACKEND_NAME}");
...@@ -3607,6 +3616,7 @@ TEST(${BACKEND_NAME}, one_hot_vector_1_far_oob) ...@@ -3607,6 +3616,7 @@ TEST(${BACKEND_NAME}, one_hot_vector_1_far_oob)
TEST(${BACKEND_NAME}, one_hot_matrix_0) TEST(${BACKEND_NAME}, one_hot_matrix_0)
{ {
SKIP_TEST_FOR("CPU", "${BACKEND_NAME}");
Shape shape_a{3, 3}; Shape shape_a{3, 3};
auto A = make_shared<op::Parameter>(element::i32, shape_a); auto A = make_shared<op::Parameter>(element::i32, shape_a);
Shape shape_r{3, 3, 3}; Shape shape_r{3, 3, 3};
...@@ -3634,6 +3644,7 @@ TEST(${BACKEND_NAME}, one_hot_matrix_0) ...@@ -3634,6 +3644,7 @@ TEST(${BACKEND_NAME}, one_hot_matrix_0)
TEST(${BACKEND_NAME}, one_hot_vector_1_fp) TEST(${BACKEND_NAME}, one_hot_vector_1_fp)
{ {
SKIP_TEST_FOR("CPU", "${BACKEND_NAME}");
Shape shape_a{8}; Shape shape_a{8};
auto A = make_shared<op::Parameter>(element::f32, shape_a); auto A = make_shared<op::Parameter>(element::f32, shape_a);
Shape shape_r{8, 3}; Shape shape_r{8, 3};
...@@ -3655,6 +3666,7 @@ TEST(${BACKEND_NAME}, one_hot_vector_1_fp) ...@@ -3655,6 +3666,7 @@ TEST(${BACKEND_NAME}, one_hot_vector_1_fp)
TEST(${BACKEND_NAME}, one_hot_vector_1_fp_nonint) TEST(${BACKEND_NAME}, one_hot_vector_1_fp_nonint)
{ {
SKIP_TEST_FOR("CPU", "${BACKEND_NAME}");
SKIP_TEST_FOR("GPU", "${BACKEND_NAME}"); SKIP_TEST_FOR("GPU", "${BACKEND_NAME}");
Shape shape_a{8}; Shape shape_a{8};
auto A = make_shared<op::Parameter>(element::f32, shape_a); auto A = make_shared<op::Parameter>(element::f32, shape_a);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment