Add new kernel config for AMD GPUs

Author: jerrymannil

torch/_inductor/kernel/mm_common.py

@@ -136,6 +136,7 @@ def filtered_configs(
         {"config": (128, 64, 32, 4, 8), "cond": True},
         {"config": (128, 128, 32, 2, 8), "cond": True},
         {"config": (128, 128, 32, 3, 4), "cond": True},
+        {"config": (128, 128, 64, 2, 8), "cond": torch.version.hip is not None},
         {"config": (128, 128, 64, 3, 4), "cond": True},
         {"config": (128, 128, 64, 5, 8), "cond": True},
     ]
@@ -165,6 +166,7 @@ def filtered_configs(
     # {"config": (32, 32, 16, 1, 2), "cond": True},
     {"config": (128, 256, 128, 3, 8), "cond": torch.version.hip is None},
     {"config": (256, 128, 128, 3, 8), "cond": torch.version.hip is None},
+    {"config": (128, 128, 64, 2, 8), "cond": torch.version.hip is not None},
 ]
 
 # Mixed precision kernel configs for small sizes of m for mm's like (16, 8192) x (8192, 8192).