Skip to content

Commit 0598a4b

Browse files
committed
fix ci
1 parent 3667578 commit 0598a4b

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

paddle/fluid/operators/detail/sendrecvop_utils.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ limitations under the License. */
1414

1515
#include "paddle/fluid/operators/detail/sendrecvop_utils.h"
1616

17+
#include <nccl.h>
1718
#include <sys/time.h>
1819
#include <thread> // NOLINT
1920

python/paddle/fluid/framework.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -487,7 +487,7 @@ def find_name(var_list, name):
487487
'rnn_memory_helper_grad', 'conditional_block', 'while', 'send',
488488
'recv', 'listen_and_serv', 'parallel_do', 'save_combine',
489489
'load_combine', 'ncclInit', 'channel_create', 'channel_close',
490-
'channel_send', 'channel_recv', 'select'
490+
'channel_send', 'channel_recv', 'select', 'gen_nccl_id'
491491
}
492492
if type not in no_kernel_op_set:
493493
self.desc.infer_var_type(self.block.desc)

0 commit comments

Comments
 (0)