3 test cases for backward spatial single and multiple

ROCm · Dec 19, 2024 · 70c28bc · 70c28bc
1 parent 6cf2959
commit 70c28bc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/test/gtest/bn_test_data.hpp b/test/gtest/bn_test_data.hpp
@@ -93,7 +93,10 @@ 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},
+        {128, 256, 14, 14, miopenBNSpatial, miopen::batchnorm::Direction::Backward, 0, 1},
+        {128, 256, 16, 16, miopenBNSpatial, miopen::batchnorm::Direction::Backward, 0, 1},
+        {670, 1, 224, 224, miopenBNSpatial, miopen::batchnorm::Direction::Backward, 0, 1}};
 }
 
 template <>