leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

More clean up

Browse Source
This commit is contained in:
KeDengMS
2021-04-18 04:29:20 +00:00
parent b7e43f5eb9
commit 83036ed646
2 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/unit/epilogue/thread/linear_combination.cu
View File

@@ -148,14 +148,12 @@ TEST(Epilogue_thread_linear_combination_gelu, device_side_f16_f16_ptr) {
}
cutlass::Array<ElementOutput, kCount> destination = linear_combination_op(accum, accum);
const float sqrt2 = sqrtf(2.0f);
cutlass::epilogue::thread::GELU<ElementOutput> gelu_func;
for (int i = 0; i < kCount; ++i) {
ElementOutput expected = gelu_func(accum[i]);
ElementOutput got = destination[i];
ElementOutput diff(fabs((float)(expected - got)));
EXPECT_TRUE(diff <= std::numeric_limits<cutlass::half_t>::epsilon());
EXPECT_TRUE(expected == got);
}
}