Skip to content

Commit 5df0d22

Browse files
committed
Revert "CUDA: fix compilation on CC 6.0 (ggml-org#16091)"
This reverts commit 368560a.
1 parent 77d1ea5 commit 5df0d22

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ggml/src/ggml-cuda/fattn-tile.cu

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ static int fattn_tile_get_kq_stride_host(const int D, const int ncols, const int
3535
switch (D) {
3636
case 64:
3737
case 128:
38+
return 128;
3839
case 256:
3940
return ncols <= 16 ? 128 : 64;
4041
default:
@@ -85,6 +86,7 @@ static constexpr __device__ int fattn_tile_get_kq_stride_device(int D, int ncols
8586
switch (D) {
8687
case 64:
8788
case 128:
89+
return 128;
8890
case 256:
8991
return ncols <= 16 ? 128 : 64;
9092
default:

0 commit comments

Comments
 (0)