Skip to content

Commit 6a5cdad

Browse files
try MUSA fix
1 parent ab20aa9 commit 6a5cdad

File tree

1 file changed

+3
-0
lines changed
  • ggml/src/ggml-cuda/vendors

1 file changed

+3
-0
lines changed

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

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
#include <musa_runtime.h>
44
#include <musa.h>
55
#include <mublas.h>
6+
#include <musa_bf16.h>
67
#include <musa_fp16.h>
78
#define CUBLAS_COMPUTE_16F CUDA_R_16F
89
#define CUBLAS_COMPUTE_32F CUDA_R_32F
@@ -132,3 +133,5 @@
132133
#define cudaKernelNodeParams musaKernelNodeParams
133134
#define cudaStreamCaptureModeRelaxed musaStreamCaptureModeRelaxed
134135
#define cudaStreamEndCapture musaStreamEndCapture
136+
137+
typedef musa_bfloat16 nv_bfloat16;

0 commit comments

Comments
 (0)