simplify

Author: xuzhao9

tritonbench/operators/Mlp/operator.py

@@ -36,18 +36,18 @@ def __init__(
 
         self.fc1 = linear_layer(in_features, hidden_features, bias=bias[0])
         self.act = act_layer()
-        self.drop1 = nn.Dropout(drop_probs[0])
-        self.norm = norm_layer(hidden_features) if norm_layer is not None else nn.Identity()
-        self.fc2 = linear_layer(hidden_features, out_features, bias=bias[1])
-        self.drop2 = nn.Dropout(drop_probs[1])
+        # self.drop1 = nn.Dropout(drop_probs[0])
+        # self.norm = norm_layer(hidden_features) if norm_layer is not None else nn.Identity()
+        # self.fc2 = linear_layer(hidden_features, out_features, bias=bias[1])
+        # self.drop2 = nn.Dropout(drop_probs[1])
 
     def forward(self, x):
         x = self.fc1(x)
         x = self.act(x)
-        x = self.drop1(x)
-        x = self.norm(x)
-        x = self.fc2(x)
-        x = self.drop2(x)
+        # x = self.drop1(x)
+        # x = self.norm(x)
+        # x = self.fc2(x)
+        # x = self.drop2(x)
         return x
 
 @torch.no_grad