From 0d1d45f3a96d4670c3511464f1a2c071b61a8729 Mon Sep 17 00:00:00 2001 From: Oleg Goncharov <64355998+Oleg-Goncharov@users.noreply.github.com> Date: Thu, 30 Jan 2025 16:24:43 +0100 Subject: [PATCH] Update test_cast_mxfp8.cu Signed-off-by: Oleg Goncharov <64355998+Oleg-Goncharov@users.noreply.github.com> --- tests/cpp/operator/test_cast_mxfp8.cu | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/cpp/operator/test_cast_mxfp8.cu b/tests/cpp/operator/test_cast_mxfp8.cu index 5ad3f11828..7e3df794d1 100644 --- 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> matrix_sizes = { {1, 16}, {16, 48}, + {128, 128}, + {256, 256}, {993, 512}, - // {768, 1024}, + {768, 1024}, // {2048, 12288}, // {65536, 128}, // {16384, 6144},