diff --git a/ggml/src/ggml-cuda/mma.cuh b/ggml/src/ggml-cuda/mma.cuh index e56da4329b..48f81e4955 100644 --- 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::supported(); } static __device__ __forceinline__ int get_i(const int l) {