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 293c400 commit a48fa64Copy full SHA for a48fa64
llmc/cuda_utils.cuh
@@ -96,15 +96,15 @@ size_t sizeof_dtype(DType type) {
96
case DType::FP16:
97
return sizeof(half);
98
case DType::BF16:
99
- return sizeof(nv_bfloat16);
+ return sizeof(__nv_bfloat16);
100
default: // handle or get compiler warning
101
fprintf(stderr, "Unknown datatype\n");
102
exit(EXIT_FAILURE);
103
}
104
105
106
DType dtype_of(float* f) { return DType::FP32; }
107
-DType dtype_of(nv_bfloat16 * f) { return DType::BF16; }
+DType dtype_of(__nv_bfloat16 * f) { return DType::BF16; }
108
DType dtype_of(half * f) { return DType::FP16; }
109
110
0 commit comments