p12346870
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

回退 'Pull Request !231 : add bool type check in communication operator '

This commit is contained in:
suteng 2020-04-11 15:30:24 +08:00 committed by Gitee
parent 5141054ecd
commit a2850cae32
1 changed files with 0 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/ops/operations/comm_ops.py
View File

@ -162,8 +162,6 @@ class AllGather(PrimitiveWithInfer):
return x_shape return x_shape
def infer_dtype(self, x_dtype): def infer_dtype(self, x_dtype):
if x_dtype == mstype.bool_:
raise TypeError("AllGather does not support 'Bool' as the dtype of input!")
return x_dtype return x_dtype
def __call__(self, tensor): def __call__(self, tensor):
@ -221,8 +219,6 @@ class ReduceScatter(PrimitiveWithInfer):
return x_shape return x_shape
def infer_dtype(self, x_dtype): def infer_dtype(self, x_dtype):
if x_dtype == mstype.bool_:
raise TypeError("ReduceScatter does not support 'Bool' as the dtype of input!")
return x_dtype return x_dtype
def __call__(self, tensor): def __call__(self, tensor):
@ -280,8 +276,6 @@ class Broadcast(PrimitiveWithInfer):
return x_shape return x_shape
def infer_dtype(self, x_dtype): def infer_dtype(self, x_dtype):
if x_dtype == mstype.bool_:
raise TypeError("Broadcast does not support 'Bool' as the dtype of input!")
return x_dtype return x_dtype
@ -324,8 +318,6 @@ class _AlltoAll(PrimitiveWithInfer):
return x_shape return x_shape
def infer_dtype(self, x_dtype): def infer_dtype(self, x_dtype):
if x_dtype == mstype.bool_:
raise TypeError("AlltoAll does not support 'Bool' as the dtype of input!")
return x_dtype return x_dtype
def __call__(self, tensor): def __call__(self, tensor):