Rename and fix 41_Gemm_BatchNorm_GELU_GroupNorm_Mean_ReLU.py → 41_Gemm_BatchNorm_GELU_ReLU.py

Author: nataliakokoromyti

KernelBench/level2/41_Gemm_BatchNorm_GELU_GroupNorm_Mean_ReLU.py renamed to KernelBench/level2/41_Gemm_BatchNorm_GELU_ReLU.py

@@ -3,13 +3,12 @@
 
 class Model(nn.Module):
     """
-    Model that performs a GEMM, BatchNorm, GELU, GroupNorm, Mean, and ReLU operations in sequence.
+    Model that performs a GEMM, BatchNorm, GELU, and ReLU in sequence.
     """
     def __init__(self, in_features, out_features, num_groups):
         super(Model, self).__init__()
         self.gemm = nn.Linear(in_features, out_features)
         self.batch_norm = nn.BatchNorm1d(out_features)
-        self.group_norm = nn.GroupNorm(num_groups, out_features)
 
     def forward(self, x):
         """
@@ -21,18 +20,16 @@ def forward(self, x):
         x = self.gemm(x)
         x = self.batch_norm(x)
         x = torch.nn.functional.gelu(x)
-        x = self.group_norm(x)
-        x = torch.mean(x, dim=1, keepdim=True)
         x = torch.relu(x)
         return x
 
 batch_size = 128
 in_features = 512
 out_features = 1024
-num_groups = 8
+num_groups = 8  # Not used anymore
 
 def get_inputs():
     return [torch.randn(batch_size, in_features)]
 
 def get_init_inputs():
-    return [in_features, out_features, num_groups]
+    return [in_features, out_features, num_groups]