Commit 63057c9f authored by Jakub Piasecki's avatar Jakub Piasecki
Browse files

fix tests in CI

parent 9004f6a5
......@@ -34,8 +34,7 @@ class TestGemmAddFastgelu : public TestGemmAdd<Tuple>
decltype(ProfileGemmAddFastgeluImpl) GetImpl() override { return ProfileGemmAddFastgeluImpl; }
};
using KernelTypes = ::testing::Types<std::tuple<F16, F16, F32, F16, F16, Row, Row, Row, Row>,
std::tuple<F16, I8, F32, F16, F16, Row, Row, Row, Row>,
using KernelTypes = ::testing::Types<std::tuple<F16, I8, F32, F16, F16, Row, Row, Row, Row>,
std::tuple<BF16, I8, F32, BF16, BF16, Row, Row, Row, Row>>;
TYPED_TEST_SUITE(TestGemmAddFastgelu, KernelTypes);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment