CINN
CINN copied to clipboard
Check fusion result
本PR主要是增加对fusion kernel的计算结果的校验功能。 通过设置FLAGS_cinn_check_fusion_pass=True,打开此功能。 具体实现如下: 一次为fusion group中的基础算子生成代码,然后分别执行单个算子和fusion算子,最后对比计算结果是否一致。
注:该功能只检验fusion算子,同时依赖基础算子生成的结算结果的正确性。如果基础算子生成代码错误,该检测无法生效。