add adhoc padding test for atten

cde398c4 · Anthony Chang · c701b3ff · cde398c4
Commit cde398c4 authored Aug 31, 2022 by Anthony Chang
Hide whitespace changes
Inline Side-by-side

Showing with 12 additions and 0 deletions

test/batched_gemm_softmax_gemm/test_batched_gemm_softmax_gemm_fp16.cpp ...gemm_softmax_gemm/test_batched_gemm_softmax_gemm_fp16.cpp +12 -0

No files found.
--- a/test/batched_gemm_softmax_gemm/test_batched_gemm_softmax_gemm_fp16.cpp
+++ b/test/batched_gemm_softmax_gemm/test_batched_gemm_softmax_gemm_fp16.cpp
@@ -37,3 +37,15 @@ TYPED_TEST(TestBatchedGemmSoftmaxGemmFP16, DISABLED_Bench_FP16)
    this->verify_ = false;
    this->Run();
 }
+TYPED_TEST(TestBatchedGemmSoftmaxGemmFP16, AdhocTest)
+{
+    this->lengths_ = std::vector<std::vector<int>>{
+        {49, 49, 64, 64, 24},
+        {64, 49, 64, 64, 24},
+        {1020, 1020, 64, 128, 24},
+        {576, 576, 64, 64, 24},
+    };
+    this->bench_ = true;
+    this->Run();
+}