未验证 提交 a01b20dd 编写于 作者: G gouzil 提交者: GitHub

[test]mv fluid op mkldnn to test/cpp/fluid/mkldnn (#53458)

上级 fca8595e
...@@ -175,12 +175,6 @@ endif() ...@@ -175,12 +175,6 @@ endif()
set(GLOB_OP_LIB ${OP_LIBRARY} CACHE INTERNAL "Global OP library") set(GLOB_OP_LIB ${OP_LIBRARY} CACHE INTERNAL "Global OP library")
if(WITH_MKLDNN)
include(mkldnn/inplace_op_tests.cmake)
include(mkldnn/caching_tests.cmake)
include(mkldnn/nhwc_op_tests.cmake)
endif()
if(WITH_UNITY_BUILD) if(WITH_UNITY_BUILD)
# Using Unity Build to compile operators, `register_operator` will cause # Using Unity Build to compile operators, `register_operator` will cause
# the unity library to lose some symbols. # the unity library to lose some symbols.
......
cc_test(
test_mkldnn_op_inplace
SRCS mkldnn/test_mkldnn_op_inplace.cc
DEPS op_registry
elementwise_add_op
activation_op
softmax_op
softmax
scope
device_context
enforce
executor)
cc_test_old(
test_mkldnn_op_nhwc
SRCS
mkldnn/test_mkldnn_op_nhwc.cc
DEPS
op_registry
pool_op
shape_op
crop_op
activation_op
generated_op
pooling
transpose_op
fused_transpose_op
scope
device_context
enforce
executor)
...@@ -11,9 +11,9 @@ if(WITH_LITE) ...@@ -11,9 +11,9 @@ if(WITH_LITE)
add_subdirectory(lite) add_subdirectory(lite)
endif() endif()
add_subdirectory(math) add_subdirectory(math)
# if(WITH_MKLDNN) if(WITH_MKLDNN)
# add_subdirectory(mkldnn) add_subdirectory(mkldnn)
# endif() endif()
add_subdirectory(nccl) add_subdirectory(nccl)
if(WITH_PSCORE) if(WITH_PSCORE)
add_subdirectory(pscore) add_subdirectory(pscore)
......
cc_test(
test_mkldnn_op_inplace
SRCS test_mkldnn_op_inplace.cc
DEPS op_registry
elementwise_add_op
activation_op
softmax_op
softmax
scope
device_context
enforce
executor)
set(TEST_MKLDNN_CACHING_DEPS set(TEST_MKLDNN_CACHING_DEPS
op_registry op_registry
elementwise_mul_op elementwise_mul_op
...@@ -16,5 +29,24 @@ if(WITH_GPU OR WITH_ROCM) ...@@ -16,5 +29,24 @@ if(WITH_GPU OR WITH_ROCM)
endif() endif()
cc_test( cc_test(
test_mkldnn_caching test_mkldnn_caching
SRCS mkldnn/test_mkldnn_caching.cc SRCS test_mkldnn_caching.cc
DEPS ${TEST_MKLDNN_CACHING_DEPS}) DEPS ${TEST_MKLDNN_CACHING_DEPS})
cc_test_old(
test_mkldnn_op_nhwc
SRCS
test_mkldnn_op_nhwc.cc
DEPS
op_registry
pool_op
shape_op
crop_op
activation_op
generated_op
pooling
transpose_op
fused_transpose_op
scope
device_context
enforce
executor)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册