Skip to content

Commit f6f5cda

Browse files
authored
Merge pull request #12555 from sneaxiy/refine_layer_norm
Refine layer_norm op
2 parents 9d6243b + c50c537 commit f6f5cda

File tree

2 files changed

+506
-2
lines changed

2 files changed

+506
-2
lines changed

paddle/fluid/operators/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -273,9 +273,9 @@ op_library(squeeze_op DEPS reshape_op)
273273
op_library(extract_rows_op DEPS memory)
274274
op_library(flatten_op DEPS reshape_op)
275275

276-
277276
if (WITH_GPU)
278277
op_library(conv_op DEPS vol2col depthwise_conv im2col)
278+
op_library(layer_norm_op DEPS cub)
279279
else()
280280
op_library(conv_op DEPS vol2col im2col)
281281
endif()

0 commit comments

Comments
 (0)