[XPU] fix fleet unittests

[houj04]

PR Category

Custom Device

PR Types

Bug fixes

Description

修复了一批分布式单测在XPU下面无法正常运行问题。一部分是有些if-elif-else语句，仅判断了custom_device然后就走GPU兜底，另一部分是有些特性XPU下面不支持或者有很细微的精度误差（atol设置为1e-8）。