Update test_cast_mxfp8.cu

Signed-off-by: Oleg Goncharov <[email protected]>
NVIDIA · Jan 30, 2025 · 0d1d45f · 0d1d45f
1 parent 1868c1a
commit 0d1d45f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/cpp/operator/test_cast_mxfp8.cu b/tests/cpp/operator/test_cast_mxfp8.cu
@@ -432,8 +432,10 @@ void performTest_x2(const ProcessingMethod processing_method,
 std::vector<std::pair<size_t, size_t>> matrix_sizes = {
     {1, 16},
     {16, 48},
+    {128, 128},
+    {256, 256},
     {993, 512},
-    // {768, 1024},
+    {768, 1024},
     // {2048, 12288},
     // {65536, 128},
     // {16384, 6144},