jasder
/
mindspore
forked from OSSInnovation/mindspore
0
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!8081 add float16 check for gpu fakequant op 

Merge pull request !8081 from yuchaojie/quant2
This commit is contained in:
mindspore-ci-bot 2020-10-31 18:34:07 +08:00 committed by Gitee
parent 1014774ab6 033e73ef12
commit fc5a3b7d97
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
mindspore/ops/operations/_quant_ops.py
View File

@ -467,7 +467,10 @@ class FakeQuantPerLayer(PrimitiveWithInfer):
return x_shape
def infer_dtype(self, x_type, min_type, max_type):
valid_types = (mstype.float16, mstype.float32)
if context.get_context('device_target') == "GPU":
valid_types = (mstype.float32,)
else:
valid_types = (mstype.float16, mstype.float32)
validator.check_tensor_type_same({"x": x_type}, valid_types, self.name)
validator.check_tensor_type_same(
{"min": min_type}, valid_types, self.name)
@ -521,7 +524,10 @@ class FakeQuantPerLayerGrad(PrimitiveWithInfer):
return dout_shape
def infer_dtype(self, dout_type, x_type, min_type, max_type):
valid_types = (mstype.float16, mstype.float32)
if context.get_context('device_target') == "GPU":
valid_types = (mstype.float32,)
else:
valid_types = (mstype.float16, mstype.float32)
validator.check_tensor_type_same(
{"dout": dout_type}, valid_types, self.name)
validator.check_tensor_type_same({"x": x_type}, valid_types, self.name)
@ -616,7 +622,10 @@ class FakeQuantPerChannel(PrimitiveWithInfer):
return x_shape
def infer_dtype(self, x_type, min_type, max_type):
valid_types = (mstype.float16, mstype.float32)
if context.get_context('device_target') == "GPU":
valid_types = (mstype.float32,)
else:
valid_types = (mstype.float16, mstype.float32)
validator.check_tensor_type_same({"x": x_type}, valid_types, self.name)
validator.check_tensor_type_same(
{"min": min_type}, valid_types, self.name)
@ -670,7 +679,10 @@ class FakeQuantPerChannelGrad(PrimitiveWithInfer):
return dout_shape
def infer_dtype(self, dout_type, x_type, min_type, max_type):
valid_types = (mstype.float16, mstype.float32)
if context.get_context('device_target') == "GPU":
valid_types = (mstype.float32,)
else:
valid_types = (mstype.float16, mstype.float32)
validator.check_tensor_type_same(
{"dout": dout_type}, valid_types, self.name)
validator.check_tensor_type_same({"x": x_type}, valid_types, self.name)