Use the correct grid_size

de7bab05 · Paul · ed22ba4f · de7bab05
Commit de7bab05 authored Oct 25, 2022 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/targets/gpu/jit/ck_gemm.cpp src/targets/gpu/jit/ck_gemm.cpp +1 -1

No files found.
--- a/src/targets/gpu/jit/ck_gemm.cpp
+++ b/src/targets/gpu/jit/ck_gemm.cpp
@@ -106,7 +106,7 @@ struct instance
    std::size_t get_grid_size(const std::array<std::size_t, 3>& config) const
    {
-        return int_div_ceil(config[0], get_pb(1)) * int_div_ceil(config[1], get_pb(1));
+        return int_div_ceil(config[0], get_pb(0)) * int_div_ceil(config[1], get_pb(1));
    }
    void set_ds_layout(const std::string& s)