op unittest for comparison op: equal/not_equal/greater_than/less_than/greater_equal/less_equal

[zzk0]

描述

From #1378

给 equal/not_equal/greater_than/less_than/greater_equal/less_equal 算子添加单元测试。

算子类型

• ElementWise：输入张量索引和输出张量索引之间存在一对一的对应关系
• Broadcast：输入张量索引和输出张量索引之间存在一对多的对应关系
• Injective：单射算子，可以将一个输出 axis 映射到一个输入 axis
• Reduction：输入张量索引和输出张量索引之间存在多对一的对应关系
• OutFusible：复杂算子，仍然可以将一对一的算子融合到其输出中。
• kNonFusible：无法融合的算子

Test Cases Checklist

张量维度

• 1D 张量
• 2D 张量
• 3D 张量
• 4D 张量

special shape

挑选 2D/3D/4D 张量中的一个，测试下面的特殊情况。

• 其中一个维度为 1
• 其中一个维度小于 1024
• 其中一个维度大于 1024
• 向量的所有维度都是 1

张量数据类型

• bool
• int32
• int64
• float16
• float32
• float64

广播

• 这个算子是否支持广播？
• 广播的测试样例

算子属性

无。

[paddle-bot]

Thanks for your contribution!

[zzk0]

这组单测会增强约 300 个测试用例，因为有六种比较算子，每种算子需要测试多种数据类型和 shape，本地耗时约 200s。

cinn-ci 运行耗时为 Total Test time (real) = 410.39 sec
cinn-ci-X86 运行耗时为 Total Test time (real) = 84.63 sec
cinn-gpu-cudnn-off 运行耗时为 Total Test time (real) = 393.43 sec

合并 PR 前后的 ci 总体运行耗时相当。

[thisjiang]

LGTM