Skip to content

Commit fc4a3a8

Browse files
committed
fix(mars): use HPCC bfloat/fp16 headers instead of MACA
Mars links the HPCC toolkit, so TypeMap must include hpcc_bfloat16.h and alias cuda_bfloat16 to __hpcc_bfloat16.
1 parent 0cdbb16 commit fc4a3a8

1 file changed

Lines changed: 7 additions & 3 deletions

File tree

‎src/native/cuda/mars/data_type_.h‎

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,23 +1,27 @@
11
#ifndef INFINI_RT_MARS_DATA_TYPE__H_
22
#define INFINI_RT_MARS_DATA_TYPE__H_
33

4-
#include <common/maca_bfloat16.h>
5-
#include <common/maca_fp16.h>
4+
#include <common/hpcc_bfloat16.h>
5+
#include <common/hpcc_fp16.h>
66
#include <hcr/hc_runtime.h>
77

88
#include "data_type.h"
99
#include "native/cuda/mars/device_.h"
1010

1111
namespace infini::rt {
1212

13+
using cuda_bfloat16 = __hpcc_bfloat16;
14+
15+
using cuda_bfloat162 = __hpcc_bfloat162;
16+
1317
template <>
1418
struct TypeMap<Device::Type::kMars, DataType::kFloat16> {
1519
using type = __half;
1620
};
1721

1822
template <>
1923
struct TypeMap<Device::Type::kMars, DataType::kBFloat16> {
20-
using type = __maca_bfloat16;
24+
using type = __hpcc_bfloat16;
2125
};
2226

2327
} // namespace infini::rt

0 commit comments

Comments
 (0)