From 82756e528229edaf14be0c6f5a0d38c159575d05 Mon Sep 17 00:00:00 2001
From: xinlipn <xinlipn@gmail.com>
Date: Sun, 15 Dec 2024 21:32:21 -0800
Subject: [PATCH 1/3] Add fwd training data for fwd spatial multiple variant =
 2 and bfp32

---
 test/gtest/bn_test_data.hpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/test/gtest/bn_test_data.hpp b/test/gtest/bn_test_data.hpp
index d3b1c6b073..65631cd36a 100644
--- a/test/gtest/bn_test_data.hpp
+++ b/test/gtest/bn_test_data.hpp
@@ -93,7 +93,9 @@ inline std::vector<BNTestCase> NetworkLarge()
         {64, 64, 112, 112, miopenBNSpatial, miopen::batchnorm::Direction::ForwardInference, 1, 0},
         {64, 64, 56, 56, miopenBNSpatial, miopen::batchnorm::Direction::Backward, 0, 1},
         {64, 64, 56, 56, miopenBNSpatial, miopen::batchnorm::Direction::ForwardTraining, 1, 1},
-        {64, 64, 56, 56, miopenBNSpatial, miopen::batchnorm::Direction::ForwardInference, 1, 0}};
+        {64, 64, 56, 56, miopenBNSpatial, miopen::batchnorm::Direction::ForwardInference, 1, 0},
+        {832, 1, 14, 14, miopenBNSpatial, miopen::batchnorm::Direction::ForwardTraining, 1, 1},
+        {832, 1, 28, 28, miopenBNSpatial, miopen::batchnorm::Direction::ForwardTraining, 1, 1}};
 }
 
 template <>

From 310192d5f6ede7a788199365477efe5fe3dd5ffd Mon Sep 17 00:00:00 2001
From: xinlipn <xinlipn@gmail.com>
Date: Tue, 17 Dec 2024 23:56:45 -0800
Subject: [PATCH 2/3] Add more test cases for forward spatial

---
 test/gtest/bn_test_data.hpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/test/gtest/bn_test_data.hpp b/test/gtest/bn_test_data.hpp
index 65631cd36a..cb474f95e7 100644
--- a/test/gtest/bn_test_data.hpp
+++ b/test/gtest/bn_test_data.hpp
@@ -94,6 +94,8 @@ inline std::vector<BNTestCase> NetworkLarge()
         {64, 64, 56, 56, miopenBNSpatial, miopen::batchnorm::Direction::Backward, 0, 1},
         {64, 64, 56, 56, miopenBNSpatial, miopen::batchnorm::Direction::ForwardTraining, 1, 1},
         {64, 64, 56, 56, miopenBNSpatial, miopen::batchnorm::Direction::ForwardInference, 1, 0},
+        {768, 1, 14, 14, miopenBNSpatial, miopen::batchnorm::Direction::ForwardTraining, 1, 1},
+        {768, 1, 23, 23, miopenBNSpatial, miopen::batchnorm::Direction::ForwardTraining, 1, 1},
         {832, 1, 14, 14, miopenBNSpatial, miopen::batchnorm::Direction::ForwardTraining, 1, 1},
         {832, 1, 28, 28, miopenBNSpatial, miopen::batchnorm::Direction::ForwardTraining, 1, 1}};
 }

From 94a7fd7fb234aa33cb4074c91d11ee86b8e31328 Mon Sep 17 00:00:00 2001
From: xinlipn <xinlipn@gmail.com>
Date: Wed, 18 Dec 2024 09:54:11 -0800
Subject: [PATCH 3/3] Replaced NetworkSmall with NetworkLarge for
 GPU_BN_FWD_Train_Large_FP64

---
 test/gtest/bn_fwd_train.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/gtest/bn_fwd_train.cpp b/test/gtest/bn_fwd_train.cpp
index 9b4722aaf8..68cb42824b 100644
--- a/test/gtest/bn_fwd_train.cpp
+++ b/test/gtest/bn_fwd_train.cpp
@@ -144,7 +144,7 @@ INSTANTIATE_TEST_SUITE_P(Smoke,
 
 INSTANTIATE_TEST_SUITE_P(Smoke,
                          GPU_BN_FWD_Train_Large_FP64,
-                         testing::Combine(testing::ValuesIn(NetworkSmall<BNTestCase>()),
+                         testing::Combine(testing::ValuesIn(NetworkLarge<BNTestCase>()),
                                           testing::ValuesIn({miopenTensorNCHW, miopenTensorNHWC}),
                                           testing::ValuesIn({testBNAPIV2})),
                          TestNameGenerator());