Skip to content

Commit bf318b8

Browse files
committed
add a fusion op: fused_residual_dropout_bias
1 parent 9d4f00b commit bf318b8

File tree

3 files changed

+1004
-0
lines changed

3 files changed

+1004
-0
lines changed

paddle/fluid/operators/fused/CMakeLists.txt

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,4 +71,9 @@ if (WITH_GPU OR WITH_ROCM)
7171
op_library(fused_bn_add_activation_op)
7272
file(APPEND ${pybind_file} "USE_CUDA_ONLY_OP(fused_bn_add_activation);\n")
7373
endif()
74+
# fused_dropout
75+
# only support CUDA
76+
if(NOT WITH_ROCM)
77+
nv_test(test_fused_residual_dropout_bias SRCS test_fused_residual_dropout_bias.cu DEPS tensor op_registry elementwise_add_op dropout_op device_context generator)
78+
endif()
7479
endif()

0 commit comments

Comments
 (0)