diff --git a/src/native/cuda/mars/data_type_.h b/src/native/cuda/mars/data_type_.h index 5af956a..2a15ce5 100644 --- a/src/native/cuda/mars/data_type_.h +++ b/src/native/cuda/mars/data_type_.h @@ -1,8 +1,8 @@ #ifndef INFINI_RT_MARS_DATA_TYPE__H_ #define INFINI_RT_MARS_DATA_TYPE__H_ -#include -#include +#include +#include #include #include "data_type.h" @@ -10,6 +10,10 @@ namespace infini::rt { +using cuda_bfloat16 = __hpcc_bfloat16; + +using cuda_bfloat162 = __hpcc_bfloat162; + template <> struct TypeMap { using type = __half; @@ -17,7 +21,7 @@ struct TypeMap { template <> struct TypeMap { - using type = __maca_bfloat16; + using type = __hpcc_bfloat16; }; } // namespace infini::rt