wehub-resource-sync
32 lines
973 B
CMake
32 lines
973 B
CMake
if(WITH_GPU OR WITH_ROCM)
|
|
# fusion_group
|
|
if(NOT APPLE AND NOT WIN32)
|
|
paddle_test(test_fusion_group_op SRCS fusion_group_op_test.cc)
|
|
endif()
|
|
if(NOT WITH_ROCM)
|
|
nv_test(
|
|
test_fused_residual_dropout_bias
|
|
SRCS fused_residual_dropout_bias_test.cu
|
|
DEPS tensor op_registry dropout_op generated_op phi common)
|
|
nv_test(
|
|
test_fused_dropout_act_bias
|
|
SRCS fused_dropout_act_bias_test.cu
|
|
DEPS tensor op_registry dropout_op generated_op phi common)
|
|
nv_test(
|
|
test_fused_layernorm_residual_dropout_bias
|
|
SRCS fused_layernorm_residual_dropout_bias_test.cu
|
|
DEPS tensor
|
|
op_registry
|
|
dropout_op
|
|
generated_op
|
|
phi
|
|
common
|
|
${CINN_DEPS})
|
|
nv_test(
|
|
test_cudnn_norm_conv
|
|
SRCS cudnn_norm_conv_test.cc
|
|
DEPS generated_op tensor op_registry phi common)
|
|
paddle_test(test_cudnn_bn_add_relu SRCS cudnn_bn_add_relu_test.cc)
|
|
endif()
|
|
endif()
|