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 ab20aa9 commit 6a5cdadCopy full SHA for 6a5cdad
ggml/src/ggml-cuda/vendors/musa.h
@@ -3,6 +3,7 @@
3
#include <musa_runtime.h>
4
#include <musa.h>
5
#include <mublas.h>
6
+#include <musa_bf16.h>
7
#include <musa_fp16.h>
8
#define CUBLAS_COMPUTE_16F CUDA_R_16F
9
#define CUBLAS_COMPUTE_32F CUDA_R_32F
@@ -132,3 +133,5 @@
132
133
#define cudaKernelNodeParams musaKernelNodeParams
134
#define cudaStreamCaptureModeRelaxed musaStreamCaptureModeRelaxed
135
#define cudaStreamEndCapture musaStreamEndCapture
136
+
137
+typedef musa_bfloat16 nv_bfloat16;
0 commit comments