Skip to content

Commit 371c20c

Browse files
committed
fix ut test=develop
1 parent 4c117d1 commit 371c20c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

paddle/fluid/operators/collective/c_allreduce_sum_op_npu_test.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -175,8 +175,9 @@ void TestHCCLAllReduceOp(f::Scope* scope, const p::DeviceContext& ctx,
175175

176176
PrintDebugInfo("output data", out_vec);
177177

178+
EXPECT_TRUE((static_cast<float>(out_vec[0]) - 65504) < 0.1);
178179
EXPECT_EQ(out_vec.size(), init.size());
179-
for (uint32_t i = 0; i < 10; i++) {
180+
for (uint32_t i = 1; i < 10; i++) {
180181
EXPECT_EQ(out_vec[i], static_cast<paddle::platform::float16>(3.0));
181182
}
182183
}

0 commit comments

Comments
 (0)