We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 19596b1 commit 96db627Copy full SHA for 96db627
ggml/src/ggml-cuda/conv2d.cuh
@@ -5,6 +5,8 @@ constexpr int BS_OC = 128;
5
constexpr int BS_ICKHKW = 16;
6
constexpr int BS_NOHOW = 128;
7
8
+// supported configuration
9
+// https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#element-types-and-matrix-sizes
10
constexpr int WMMA_M = 16;
11
constexpr int WMMA_N = 16;
12
constexpr int WMMA_K = 16;
0 commit comments