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 52d9ccc commit 66eab7aCopy full SHA for 66eab7a
ggml/src/ggml-cuda/vendors/hip.h
@@ -4,6 +4,7 @@
4
#include <hip/hip_runtime.h>
5
#include <hipblas/hipblas.h>
6
#include <hip/hip_fp16.h>
7
+#include <hip/hip_bf16.h>
8
#include <hip/hip_bfloat16.h>
9
// for rocblas_initialize()
10
#include "rocblas/rocblas.h"
@@ -200,7 +201,7 @@
200
201
#endif
202
203
typedef hip_bfloat16 nv_bfloat16;
-typedef short2 nv_bfloat162; // FIXME there is no 2x BF16 type being defined in bfloat16.h, ad-hoc compilation fix
204
+typedef __hip_bfloat162 nv_bfloat162;
205
206
typedef int8_t int8x4_t __attribute__((ext_vector_type(4)));
207
typedef uint8_t uint8x4_t __attribute__((ext_vector_type(4)));
0 commit comments