【Infer Symbolic Shape No.86, 89】[buaa]Add multiplex number_count op
PR Category
CINN
PR Types
Improvements
Description
添加multiplex number_count算子符号推导接口实现。
你的PR提交成功,感谢你对开源项目的贡献! 请关注后续CI自动化测试结果,详情请参考Paddle-CI手册。 Your PR has been submitted. Thanks for your contribution! Please wait for the result of CI firstly. See Paddle CI Manual for details.
抱歉,代码中有些命名不规范的地方,比如说里面ins的变量,应该用更加意义明显的变量名。
为了不占用CI资源,等老师review完后再改吧。
抱歉,代码中有些命名不规范的地方,比如说里面
ins的变量,应该用更加意义明显的变量名。 为了不占用CI资源,等老师review完后再改吧。
好的
multiplex对应测试文件(test_multiplex_op.py)默认打开check_pir,coverage流水线通过
number_count(test_number_count_op.py)未打开,打开后通过测试
原number_count测试文件节选
class TestNumberCountOpInt64(op_test.OpTest):
def setUp(self):
upper_num = 16
self.op_type = "number_count"
self.python_api = number_count_wrapper
x = np.random.randint(-1, upper_num, size=(1000, 2)).astype('int64')
self.inputs = {'numbers': x}
self.outputs = {'Out': count(x, upper_num)}
self.attrs = {"upper_range": upper_num}
def test_forward(self):
self.check_output_with_place(paddle.CUDAPlace(0))
ce测试挂了,先修改一下上述问题,看看能不能解决。