add new unittest (#77)

# Description

Please include a brief summary of the changes, relevant motivation and
context.

Fixes # (issue)

## Type of change

- [ ] Documentation change (change only to the documentation, either a
fix or a new content)
- [ ] Bug fix (non-breaking change which fixes an issue)
- [ ] New feature (non-breaking change which adds functionality)
- [ ] Breaking change (fix or feature that would cause existing
functionality to not work as expected)
- [ ] Infra/Build change
- [ ] Code refactoring

## Changes

Please list the changes introduced in this PR:

- Change A
- Change B

# Checklist:

- [ ] I have read and followed the [contributing
guidelines](https://github.com/NVIDIA/TransformerEngine/blob/main/CONTRIBUTING.rst)
- [ ] The functionality is complete
- [ ] I have commented my code, particularly in hard-to-understand areas
- [ ] I have made corresponding changes to the documentation
- [ ] My changes generate no new warnings
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] New and existing unit tests pass locally with my changes

Author: zhaoyinglia

transformer_engine/plugin/tests/run_all_tests.py

@@ -10,6 +10,8 @@
 from test_softmax import SoftmaxTests
 from test_optimizer import OptimizerTests
 from test_flash_attention import FlashAttentionTests
+from test_te_general_grouped import grouped_gemmTests
+from test_policy import run_all_tests
 
 
 def main():
@@ -27,6 +29,7 @@ def main():
         SoftmaxTests(device=device),
         OptimizerTests(device=device),
         FlashAttentionTests(device=device),
+        grouped_gemmTests(device=device),
     ]
 
     results = []
@@ -49,6 +52,8 @@ def main():
     print(f"Total: {total_passed}/{total_tests} test suites passed")
     print("=" * 70)
 
+    run_all_tests()
+
     return 0 if all(success for _, success in results) else 1