Include 2048

ce364aae · Paul · dc030d1c · ce364aae
Commit ce364aae authored Oct 20, 2022 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/targets/gpu/fuse_ck.cpp src/targets/gpu/fuse_ck.cpp +1 -1

No files found.
--- a/src/targets/gpu/fuse_ck.cpp
+++ b/src/targets/gpu/fuse_ck.cpp
@@ -55,7 +55,7 @@ MIGRAPHX_PRED_MATCHER(is_ck_gemm, instruction_ref ins)
    auto b = ins->inputs().back()->get_shape();
    if(a.lens().size() > 2 or b.lens().size() > 2)
        return false;
-    if(a.lens()[1] >= 2048)
+    if(a.lens()[1] > 2048)
        return false;
    return (a.lens()[0] % 8 == 0 and a.lens()[1] % 8 == 0 and b.lens()[0] % 8 == 0 and
            b.lens()[1] % 8 == 0);