Commit c082fe1e authored by fenglei.tian's avatar fenglei.tian

nvrtc name in cmake, enable mkldnn only for CPU

parent 4e64e96f
...@@ -171,8 +171,8 @@ if(LLVM_INCLUDE_DIR) ...@@ -171,8 +171,8 @@ if(LLVM_INCLUDE_DIR)
endif() endif()
if(CUDA_INCLUDE_DIRS) if(CUDA_INCLUDE_DIRS)
find_library(CUDA_nvrtc_LIBRARY nvrtc /usr/local/cuda/lib64) find_library(CUDA_NVRTC_LIBRARY nvrtc /usr/local/cuda/lib64)
target_link_libraries(unit-test ${CUDA_nvrtc_LIBRARY} ${CUDA_LIBRARIES} ${CUDNN_LIBRARIES} ${CUDA_CUBLAS_LIBRARIES}) target_link_libraries(unit-test ${CUDA_NVRTC_LIBRARY} ${CUDA_LIBRARIES} ${CUDNN_LIBRARIES} ${CUDA_CUBLAS_LIBRARIES})
endif() endif()
target_link_libraries(unit-test ngraph libgtest pthread) target_link_libraries(unit-test ngraph libgtest pthread)
......
...@@ -6244,8 +6244,7 @@ TEST(${BACKEND_NAME}, convolution_outlining) ...@@ -6244,8 +6244,7 @@ TEST(${BACKEND_NAME}, convolution_outlining)
TEST(${BACKEND_NAME}, mkldnn_layouts) TEST(${BACKEND_NAME}, mkldnn_layouts)
{ {
SKIP_TEST_FOR("GPU", "${BACKEND_NAME}"); ONLY_ENABLE_TEST_FOR("CPU", "${BACKEND_NAME");
SKIP_TEST_FOR("INTERPRETER", "${BACKEND_NAME}");
Shape shape_a{1, 16, 2, 2}; Shape shape_a{1, 16, 2, 2};
auto A = make_shared<op::Parameter>(element::f32, shape_a); auto A = make_shared<op::Parameter>(element::f32, shape_a);
Shape shape_b{32, 16, 1, 1}; Shape shape_b{32, 16, 1, 1};
......
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