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

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

上级 fca8595e
......@@ -175,12 +175,6 @@ endif()
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)
# Using Unity Build to compile operators, `register_operator` will cause
# 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)
add_subdirectory(lite)
endif()
add_subdirectory(math)
# if(WITH_MKLDNN)
# add_subdirectory(mkldnn)
# endif()
if(WITH_MKLDNN)
add_subdirectory(mkldnn)
endif()
add_subdirectory(nccl)
if(WITH_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
op_registry
elementwise_mul_op
......@@ -16,5 +29,24 @@ if(WITH_GPU OR WITH_ROCM)
endif()
cc_test(
test_mkldnn_caching
SRCS mkldnn/test_mkldnn_caching.cc
SRCS test_mkldnn_caching.cc
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.
先完成此消息的编辑!
想要评论请 注册