Skip to content

Commit 96db627

Browse files
committed
CUDA: conv2d added comment
1 parent 19596b1 commit 96db627

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ggml/src/ggml-cuda/conv2d.cuh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,8 @@ constexpr int BS_OC = 128;
55
constexpr int BS_ICKHKW = 16;
66
constexpr int BS_NOHOW = 128;
77

8+
// supported configuration
9+
// https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#element-types-and-matrix-sizes
810
constexpr int WMMA_M = 16;
911
constexpr int WMMA_N = 16;
1012
constexpr int WMMA_K = 16;

0 commit comments

Comments
 (0)