35 lines
942 B
C++
35 lines
942 B
C++
#ifdef USE_C10D_UCC
|
|
#include <gtest/gtest.h>
|
|
#include <torch/csrc/distributed/c10d/UCCUtils.hpp>
|
|
|
|
#include <utility>
|
|
#include <vector>
|
|
TEST(ProcessGroupUCCTest, testTrim) {
|
|
std::vector<std::pair<std::string, std::string>> tests = {
|
|
{" allreduce ", "allreduce"},
|
|
{"\tallgather", "allgather"},
|
|
{"send\n", "send"},
|
|
};
|
|
for (auto entry : tests) {
|
|
ASSERT_EQ(c10d::trim(entry.first), entry.second);
|
|
}
|
|
}
|
|
|
|
TEST(ProcessGroupUCCTest, testToLower) {
|
|
std::vector<std::pair<std::string, std::string>> tests = {
|
|
{"AllReduce", "allreduce"},
|
|
{"ALLGATHER", "allgather"},
|
|
{"send", "send"},
|
|
};
|
|
for (auto entry : tests) {
|
|
ASSERT_EQ(c10d::tolower(entry.first), entry.second);
|
|
}
|
|
}
|
|
|
|
TEST(ProcessGroupUCCTest, testParseList) {
|
|
std::string input = "\tAllReduce, ALLGATHER, send\n";
|
|
std::vector<std::string> expect{"allreduce", "allgather", "send"};
|
|
ASSERT_EQ(c10d::parse_list(input), expect);
|
|
}
|
|
#endif
|