If you ever feel like improving your matmul kernel you can check this blog https://siboehm.com/articles/22/CUDA-MMM