fix ci

7 years ago · 0598a4b366
parent 3667578ec2
commit 0598a4b366
2 changed files with 2 additions and 1 deletions
--- a/paddle/fluid/operators/detail/sendrecvop_utils.cc
+++ b/paddle/fluid/operators/detail/sendrecvop_utils.cc
@ -14,6 +14,7 @@ limitations under the License. */

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

+#include <nccl.h>
 #include <sys/time.h>
 #include <thread>  // NOLINT

--- a/python/paddle/fluid/framework.py
+++ b/python/paddle/fluid/framework.py
@ -487,7 +487,7 @@ class Operator(object):
            'rnn_memory_helper_grad', 'conditional_block', 'while', 'send',
            'recv', 'listen_and_serv', 'parallel_do', 'save_combine',
            'load_combine', 'ncclInit', 'channel_create', 'channel_close',
-            'channel_send', 'channel_recv', 'select'
+            'channel_send', 'channel_recv', 'select', 'gen_nccl_id'
        }
        if type not in no_kernel_op_set:
            self.desc.infer_var_type(self.block.desc)