Commit 45a6b898 authored by rocking's avatar rocking
Browse files

Add more testcases

parent 19690f5b
......@@ -32,7 +32,8 @@ class TestGemmAddReluAddLayernorm : public ::testing::Test
void Run()
{
std::vector<std::vector<ck::index_t>> lengths = {{1024, 1024, 1024}};
std::vector<std::vector<ck::index_t>> lengths = {
{1024, 1024, 1024}, {2048, 640, 640}, {1, 1, 1}};
for(auto length : lengths)
{
......
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