fix typo and refine error information in CI_register_dataType (#21932)

* fix typo and refine error information test=document_fix, test=develop

* fix typo test=document_fix test=develop

* fix bug test=document_fix test=develop
release/1.7
Aurelius84 5 years ago committed by Tao Luo
parent cff7a49856
commit 15e41e9115

@ -72,7 +72,7 @@ fi
op_type_spec_diff=`python ${PADDLE_ROOT}/tools/check_op_register_type.py ${PADDLE_ROOT}/paddle/fluid/OP_TYPE_DEV.spec ${PADDLE_ROOT}/paddle/fluid/OP_TYPE_PR.spec`
if [ "$op_type_spec_diff" != "" ]; then
echo_line="More data_type of new operator should be regitered in your PR. Please make sure that both float/double (or int/int64_t) have been regitered. You must have one RD (Aurelius84 or liym27 or zhhsplendid)approval for the data_type registration of new operator.\n"
echo_line="You must have one RD (Aurelius84 (Recommend) or liym27 or zhhsplendid)approval for the data_type registration of new operator. More data_type of new operator should be registered in your PR. Please make sure that both float/double (or int/int64_t) have been registered.\n For more details, please click [https://github.com/PaddlePaddle/Paddle/wiki/Data-types-of-generic-Op-must-be-fully-registered].\n"
check_approval 1 9301846 33742067 7913861
fi

@ -25,6 +25,9 @@ import difflib
import collections
import paddle.fluid as fluid
INTS = set(['int', 'int64_t'])
FLOATS = set(['float', 'double'])
def get_all_kernels():
all_kernels_info = fluid.core._get_all_register_op_kernels()
@ -54,8 +57,15 @@ def read_file(file_path):
return content
INTS = set(['int', 'int64_t'])
FLOATS = set(['float', 'double'])
def print_diff(op_type, register_types):
lack_types = set()
if len(INTS - register_types) == 1:
lack_types |= INTS - register_types
if len(FLOATS - register_types) == 1:
lack_types |= FLOATS - register_types
print("{} only supports [{}] now, but lacks [{}].".format(op_type, " ".join(
register_types), " ".join(lack_types)))
def check_add_op_valid():
@ -72,7 +82,7 @@ def check_add_op_valid():
register_types = set(op_info[1:])
if len(FLOATS - register_types) == 1 or \
len(INTS - register_types) == 1:
print(each_diff)
print_diff(op_info[0], register_types)
if len(sys.argv) == 1:

Loading…
Cancel
Save