change k's value to a multiple of 256 (#98)

* Add new problem nvfp4_gemm to nvidia.yaml

* change k's value to a multiple of 256 for better perf (simplify some logic).

* revert unnecessary  change.

Author: vickiw973

problems/nvidia/nvfp4_group_gemm/task.yml

@@ -45,16 +45,16 @@ templates:
   Python: "template.py"
 
 tests:
-  - {"m": [96, 128], "n": [128, 256], "k": [128, 512], "g": 2, "seed": 1111}
+  - {"m": [96, 128], "n": [128, 256], "k": [256, 512], "g": 2, "seed": 1111}
   - {"m": [256, 72], "n": [512, 384], "k": [256, 256], "g": 2, "seed": 1111}
-  - {"m": [128, 128], "n": [128, 256], "k": [128, 512], "g": 2, "seed": 1111}
-  - {"m": [80, 128, 256], "n": [384, 256, 128], "k": [256, 512, 128], "g": 3, "seed": 1111}
-  - {"m": [64, 72, 96], "n": [128, 384, 512], "k": [384, 512, 128], "g": 3, "seed": 1111}
-  - {"m": [64, 256, 128], "n": [768, 128, 256], "k": [512, 512, 128], "g": 3, "seed": 1111}
-  - {"m": [128, 128, 64], "n": [256, 512, 512], "k": [768, 128, 768], "g": 3, "seed": 1111}
-  - {"m": [128, 128, 128, 128], "n": [128, 128, 128, 128], "k": [128, 128, 128, 128], "g": 4, "seed": 1111}
+  - {"m": [128, 128], "n": [128, 256], "k": [512, 256], "g": 2, "seed": 1111}
+  - {"m": [80, 128, 256], "n": [384, 256, 128], "k": [256, 512, 256], "g": 3, "seed": 1111}
+  - {"m": [64, 72, 96], "n": [128, 384, 512], "k": [512, 512, 256], "g": 3, "seed": 1111}
+  - {"m": [64, 256, 128], "n": [768, 128, 256], "k": [512, 256, 512], "g": 3, "seed": 1111}
+  - {"m": [128, 128, 64], "n": [256, 512, 512], "k": [768, 256, 768], "g": 3, "seed": 1111}
+  - {"m": [128, 128, 128, 128], "n": [128, 128, 128, 128], "k": [512, 256, 512, 256], "g": 4, "seed": 1111}
   - {"m": [40, 56, 384, 512], "n": [512, 384, 256, 128], "k": [256, 256, 256, 256], "g": 4, "seed": 1111}
-  - {"m": [512, 384, 256, 128], "n": [256, 256, 256, 256], "k": [512, 128, 512, 128], "g": 4, "seed": 1111}
+  - {"m": [512, 384, 256, 128], "n": [256, 256, 256, 256], "k": [512, 768, 512, 768], "g": 4, "seed": 1111}
 
 benchmarks:
   - {"m": [80, 176, 128, 72, 64, 248, 96, 160], "n": [4096, 4096, 4096, 4096, 4096, 4096, 4096, 4096], "k": [7168, 7168, 7168, 7168, 7168, 7168, 7168, 7168], "g": 8, "seed": 1111}