Skip to content

Commit dec22a7

Browse files
KyL0Njordankanter
authored andcommitted
cuda : fix compile error in jetson platform (ggml-org#4975)
* cuda: fix compile error in jetson platform * cuda: update comment in ggml-cuda.cu * cuda: update ggml-cuda.cu comment
1 parent 52322a5 commit dec22a7

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

ggml-cuda.cu

+5-3
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,6 @@
1212
#include <vector>
1313
#include <map>
1414
#include <array>
15-
#include "ggml-cuda.h"
16-
#include "ggml.h"
17-
#include "ggml-backend-impl.h"
1815

1916
#if defined(GGML_USE_HIPBLAS)
2017
#include <hip/hip_runtime.h>
@@ -118,6 +115,11 @@
118115

119116
#endif // defined(GGML_USE_HIPBLAS)
120117

118+
// ggml-cuda need half type so keep ggml headers include at last
119+
#include "ggml-cuda.h"
120+
#include "ggml.h"
121+
#include "ggml-backend-impl.h"
122+
121123
#define CUDART_HMAX 11070 // CUDA 11.7, min. ver. for which __hmax and __hmax2 are known to work (may be higher than needed)
122124

123125
#define CC_PASCAL 600

0 commit comments

Comments
 (0)