Skip to content

[Metax] mccl #2376

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
metax666 merged 2 commits into from
Jan 27, 2026
Merged

[Metax] mccl #2376

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
bd1718e
[Metax] mccl
duqimeng Jan 26, 2026
a42c522
Merge branch 'develop' into develop
metax666 Jan 27, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
29 changes: 1 addition & 28 deletions backends/metax_gpu/runtime/runtime.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,7 @@ inline ncclDataType_t PDDataTypeToNcclDataType(C_DataType type) {
return ncclFloat32;
} else if (type == C_DataType::BFLOAT16) {
return ncclBfloat16;
} else if (type == C_DataType::UINT8) {
} else if (type == C_DataType::UINT8 || type == C_DataType::BOOL) {
return ncclUint8;
} else if (type == C_DataType::UINT32) {
return ncclUint32;
Expand All @@ -93,33 +93,6 @@ inline ncclDataType_t PDDataTypeToNcclDataType(C_DataType type) {
return ncclFloat32;
}

inline mcclDataType_t PDDataTypeToMcclDataType(C_DataType type) {
if (type == C_DataType::FLOAT32) {
return mcclFloat32;
} else if (type == C_DataType::BFLOAT16) {
return mcclBfloat16;
} else if (type == C_DataType::UINT8) {
return mcclUint8;
} else if (type == C_DataType::UINT32) {
return mcclUint32;
} else if (type == C_DataType::UINT64) {
return mcclUint64;
} else if (type == C_DataType::INT8) {
return mcclInt8;
} else if (type == C_DataType::INT32) {
return mcclInt32;
} else if (type == C_DataType::INT64) {
return mcclInt64;
} else if (type == C_DataType::FLOAT16) {
return mcclFloat16;
} else if (type == C_DataType::FLOAT64) {
return mcclFloat64;
} else {
LOG(ERROR) << "Datatype " << type << " in nccl is not supported.";
}
return ncclFloat32;
}

#define NCCL_CHECK(cmd) \
do { \
ncclResult_t r = cmd; \
Expand Down