AI/rocm-systems
2
0
Fork 0
程式碼 問題 合併請求 Actions 套件 專案 版本發布 Wiki 動態

Adding better naming to unit tests for filtering; adding short and full unit test suites (#235)

瀏覽代碼
此提交包含在：
Stanley Tsang
2020-07-21 12:19:47 -06:00
提交者 GitHub
父節點 35c5a7fe45
當前提交 684f3e6af4
共有 14 個檔案被更改，包括 112 行新增 和 39 行删除
下載 Patch 檔 下載差異檔 展開所有檔案 折疊所有檔案
test/CorrectnessTest.hpp
+45
查看文件
@@ -170,6 +170,51 @@ namespace CorrectnessTests
// - Each test is instantiated with a different TestTuple
class CorrectnessTest : public testing::TestWithParam<TestTuple>
{
public:
struct PrintToStringParamName
{
std::string operator()(const testing::TestParamInfo<CorrectnessTest::ParamType>& info)
{
std::string name;
name += opStrings[std::get<0>(info.param)] + "_";
name += dataTypeStrings[std::get<1>(info.param)] + "_";
name += std::to_string(std::get<2>(info.param)) + "elements_";
name += std::to_string(std::get<3>(info.param)) + "devices_";
name += std::get<4>(info.param) == true ? "inplace_" : "outofplace_";
std::string envVars = std::string(std::get<5>(info.param));
std::replace(envVars.begin(), envVars.end(), '=', '_');
name += envVars;
return name;
}
std::map<ncclRedOp_t, std::string> opStrings
{
{ncclSum, "sum"},
{ncclProd, "prod"},
{ncclMax, "max"},
{ncclMin, "min"}
};
std::map<ncclDataType_t, std::string> dataTypeStrings
{
{ncclInt8, "int8"},
{ncclChar, "char"},
{ncclUint8, "uint8"},
{ncclInt32, "int32"},
{ncclInt, "int"},
{ncclUint32, "uint32"},
{ncclInt64, "int64"},
{ncclUint64, "uint64"},
{ncclFloat16, "float16"},
{ncclHalf, "half"},
{ncclFloat32, "float32"},
{ncclFloat64, "float64"},
{ncclDouble, "double"},
{ncclBfloat16, "bfloat16"}
};
};
protected:
// This code is called per test-tuple
test/test_AllGather.cpp
+3 -2
查看文件
@@ -88,7 +88,7 @@ namespace CorrectnessTests
}
INSTANTIATE_TEST_CASE_P(AllGatherCorrectnessSweep,
INSTANTIATE_TEST_SUITE_P(AllGatherCorrectnessSweep,
AllGatherCorrectnessTest,
testing::Combine(
// Reduction operator (not used)
@@ -110,5 +110,6 @@ namespace CorrectnessTests
testing::Values(2,3,4,5,6,7,8),
// In-place or not
testing::Values(false, true),
testing::Values("")));
testing::Values("")),
CorrectnessTest::PrintToStringParamName());
} // namespace
test/test_AllReduce.cpp
+3 -2
查看文件
@@ -36,7 +36,7 @@ namespace CorrectnessTests
dataset.Release();
}
INSTANTIATE_TEST_CASE_P(AllReduceCorrectnessSweep,
INSTANTIATE_TEST_SUITE_P(AllReduceCorrectnessSweep,
AllReduceCorrectnessTest,
testing::Combine(
// Reduction operator
@@ -58,5 +58,6 @@ namespace CorrectnessTests
testing::Values(2,3,4,5,6,7,8),
// In-place or not
testing::Values(false, true),
testing::Values("")));
testing::Values("")),
CorrectnessTest::PrintToStringParamName());
} // namespace
test/test_AllReduceAbort.cpp
+3 -2
查看文件
@@ -126,7 +126,7 @@ namespace CorrectnessTests
dataset.Release();
}
INSTANTIATE_TEST_CASE_P(AllReduceAbortSweep,
INSTANTIATE_TEST_SUITE_P(AllReduceAbortSweep,
AllReduceAbortTest,
testing::Combine(
// Reduction operator
@@ -139,5 +139,6 @@ namespace CorrectnessTests
testing::Values(2, 4),
// In-place or not
testing::Values(false),
testing::Values("")));
testing::Values("")),
CorrectnessTest::PrintToStringParamName());
} // namespace
test/test_AllToAll.cpp
+3 -2
查看文件
@@ -40,7 +40,7 @@ namespace CorrectnessTests
dataset.Release();
}
INSTANTIATE_TEST_CASE_P(AllToAllCorrectnessSweep,
INSTANTIATE_TEST_SUITE_P(AllToAllCorrectnessSweep,
AllToAllCorrectnessTest,
testing::Combine(
// Reduction operator is not used
@@ -62,5 +62,6 @@ namespace CorrectnessTests
testing::Values(2,3,4,5,6,7,8),
// In-place or not
testing::Values(false),
testing::Values("RCCL_ALLTOALL_KERNEL_DISABLE=0", "RCCL_ALLTOALL_KERNEL_DISABLE=1")));
testing::Values("RCCL_ALLTOALL_KERNEL_DISABLE=0", "RCCL_ALLTOALL_KERNEL_DISABLE=1")),
CorrectnessTest::PrintToStringParamName());
} // namespace
test/test_Broadcast.cpp
+3 -2
查看文件
@@ -44,7 +44,7 @@ namespace CorrectnessTests
dataset.Release();
}
INSTANTIATE_TEST_CASE_P(BroadcastCorrectnessSweep,
INSTANTIATE_TEST_SUITE_P(BroadcastCorrectnessSweep,
BroadcastCorrectnessTest,
testing::Combine(
// Reduction operator is not used
@@ -66,5 +66,6 @@ namespace CorrectnessTests
testing::Values(2,3,4,5,6,7,8),
// In-place or not
testing::Values(false, true),
testing::Values("")));
testing::Values("")),
CorrectnessTest::PrintToStringParamName());
} // namespace
test/test_BroadcastAbort.cpp
+3 -2
查看文件
@@ -129,7 +129,7 @@ namespace CorrectnessTests
dataset.Release();
}
INSTANTIATE_TEST_CASE_P(BroadcastAbortSweep,
INSTANTIATE_TEST_SUITE_P(BroadcastAbortSweep,
BroadcastAbortTest,
testing::Combine(
// Reduction operator
@@ -142,5 +142,6 @@ namespace CorrectnessTests
testing::Values(2, 4),
// In-place or not
testing::Values(false),
testing::Values("")));
testing::Values("")),
CorrectnessTest::PrintToStringParamName());
} // namespace
test/test_CombinedCalls.cpp
+3 -2
查看文件
@@ -74,7 +74,7 @@ namespace CorrectnessTests
}
}
INSTANTIATE_TEST_CASE_P(CombinedCallsCorrectnessSweep,
INSTANTIATE_TEST_SUITE_P(CombinedCallsCorrectnessSweep,
CombinedCallsCorrectnessTest,
testing::Combine(
// Reduction operator (not used)
@@ -96,5 +96,6 @@ namespace CorrectnessTests
testing::Values(2,3,4,5,6,7,8),
// In-place or not
testing::Values(false, true),
testing::Values("")));
testing::Values("")),
CorrectnessTest::PrintToStringParamName());
} // namespace
test/test_Gather.cpp
+3 -2
查看文件
@@ -44,7 +44,7 @@ namespace CorrectnessTests
dataset.Release();
}
INSTANTIATE_TEST_CASE_P(GatherCorrectnessSweep,
INSTANTIATE_TEST_SUITE_P(GatherCorrectnessSweep,
GatherCorrectnessTest,
testing::Combine(
// Reduction operator is not used
@@ -66,5 +66,6 @@ namespace CorrectnessTests
testing::Values(2,3,4,5,6,7,8),
// In-place or not
testing::Values(false),
testing::Values("RCCL_ALLTOALL_KERNEL_DISABLE=0", "RCCL_ALLTOALL_KERNEL_DISABLE=1")));
testing::Values("RCCL_ALLTOALL_KERNEL_DISABLE=0", "RCCL_ALLTOALL_KERNEL_DISABLE=1")),
CorrectnessTest::PrintToStringParamName());
} // namespace
test/test_GroupCalls.cpp
+5 -4
查看文件
@@ -94,9 +94,9 @@ namespace CorrectnessTests
}
}
INSTANTIATE_TEST_CASE_P(GroupCallsCorrectnessSweep,
GroupCallsCorrectnessTest,
testing::Combine(
INSTANTIATE_TEST_SUITE_P(GroupCallsCorrectnessSweep,
GroupCallsCorrectnessTest,
testing::Combine(
// Reduction operator (not used)
testing::Values(ncclSum),
// Data types
@@ -116,5 +116,6 @@ namespace CorrectnessTests
testing::Values(2,3,4,5,6,7,8),
// In-place or not
testing::Values(false, true),
testing::Values("")));
testing::Values("")),
CorrectnessTest::PrintToStringParamName());
} // namespace
test/test_Reduce.cpp
+5 -4
查看文件
@@ -44,9 +44,9 @@ namespace CorrectnessTests
dataset.Release();
}
INSTANTIATE_TEST_CASE_P(ReduceCorrectnessSweep,
ReduceCorrectnessTest,
testing::Combine(
INSTANTIATE_TEST_SUITE_P(ReduceCorrectnessSweep,
ReduceCorrectnessTest,
testing::Combine(
// Reduction operator
testing::Values(ncclSum, ncclProd, ncclMax, ncclMin),
// Data types
@@ -66,5 +66,6 @@ namespace CorrectnessTests
testing::Values(2,3,4,5,6,7,8),
// In-place or not
testing::Values(false, true),
testing::Values("")));
testing::Values("")),
CorrectnessTest::PrintToStringParamName());
} // namespace
test/test_ReduceScatter.cpp
+5 -4
查看文件
@@ -42,9 +42,9 @@ namespace CorrectnessTests
dataset.Release();
}
INSTANTIATE_TEST_CASE_P(ReduceScatterCorrectnessSweep,
ReduceScatterCorrectnessTest,
testing::Combine(
INSTANTIATE_TEST_SUITE_P(ReduceScatterCorrectnessSweep,
ReduceScatterCorrectnessTest,
testing::Combine(
// Reduction operator
testing::Values(ncclSum, ncclProd, ncclMax, ncclMin),
// Data types
@@ -64,5 +64,6 @@ namespace CorrectnessTests
testing::Values(2,3,4,5,6,7,8),
// In-place or not
testing::Values(false, true),
testing::Values("")));
testing::Values("")),
CorrectnessTest::PrintToStringParamName());
} // namespace
test/test_Scatter.cpp
+3 -2
查看文件
@@ -44,7 +44,7 @@ namespace CorrectnessTests
dataset.Release();
}
INSTANTIATE_TEST_CASE_P(ScatterCorrectnessSweep,
INSTANTIATE_TEST_SUITE_P(ScatterCorrectnessSweep,
ScatterCorrectnessTest,
testing::Combine(
// Reduction operator is not used
@@ -66,5 +66,6 @@ namespace CorrectnessTests
testing::Values(2,3,4,5,6,7,8),
// In-place or not
testing::Values(false),
testing::Values("RCCL_ALLTOALL_KERNEL_DISABLE=0", "RCCL_ALLTOALL_KERNEL_DISABLE=1")));
testing::Values("RCCL_ALLTOALL_KERNEL_DISABLE=0", "RCCL_ALLTOALL_KERNEL_DISABLE=1")),
CorrectnessTest::PrintToStringParamName());
} // namespace