Skip to content

Commit 66eab7a

Browse files
try CI fix
1 parent 52d9ccc commit 66eab7a

File tree

1 file changed

+2
-1
lines changed
  • ggml/src/ggml-cuda/vendors

1 file changed

+2
-1
lines changed

ggml/src/ggml-cuda/vendors/hip.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
#include <hip/hip_runtime.h>
55
#include <hipblas/hipblas.h>
66
#include <hip/hip_fp16.h>
7+
#include <hip/hip_bf16.h>
78
#include <hip/hip_bfloat16.h>
89
// for rocblas_initialize()
910
#include "rocblas/rocblas.h"
@@ -200,7 +201,7 @@
200201
#endif
201202

202203
typedef hip_bfloat16 nv_bfloat16;
203-
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;
204205

205206
typedef int8_t int8x4_t __attribute__((ext_vector_type(4)));
206207
typedef uint8_t uint8x4_t __attribute__((ext_vector_type(4)));

0 commit comments

Comments
 (0)