disable ncclInit infer shape & var type

7 years ago · e9ddaaba9c
parent 672cdc21a0
commit e9ddaaba9c
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/v2/fluid/framework.py
+++ b/python/paddle/v2/fluid/framework.py
@ -490,7 +490,7 @@ class Operator(object):
            'feed', 'fetch', 'save', 'load', 'recurrent',
            'rnn_memory_helper_grad', 'conditional_block', 'while', 'send',
            'recv', 'listen_and_serv', 'parallel_do', 'save_combine',
-            'load_combine'
+            'load_combine', 'ncclInit'
        }
        if type not in no_kernel_op_set:
            self.desc.infer_var_type(self.block.desc)