align i-major and j-major

2025-12-13 17:05:22 +08:00 · 2025-12-13 17:05:22 +08:00 · 8b26bc388a
parent 62e4954d3f
commit 8b26bc388a
1 changed files with 1 additions and 2 deletions
--- a/ggml/src/ggml-cuda/mma.cuh
+++ b/ggml/src/ggml-cuda/mma.cuh
@ -442,8 +442,7 @@ namespace ggml_cuda_mma {
        T x[ne] = {0};

        static constexpr __device__ bool supported() {
-            if (I == 16 && J == 16) return true;
-            return false;
+            return tile<I_, J_, T, DATA_LAYOUT_I_MAJOR>::supported();
        }

        static __device__ __forceinline__ int get_i(const int l) {