mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-24 14:54:51 +00:00
e227fdc1ef
- Modified test/mp_unit/mp_unit_tests.hpp to use ../framework.hpp instead of gtest/gtest.h - Enhanced test/framework.hpp with GTest-compatible APIs: - Added Environment base class for global test setup/teardown - Added TestInfo and UnitTest classes for test metadata access - Added GTEST_SKIP macro support via SkipHelper class - Added namespace alias 'testing' for compatibility - Added InitGoogleTest and AddGlobalTestEnvironment helper functions - Updated test/framework.cc with implementations for new classes - All mp_unit test files now use framework.hpp through mp_unit_tests.hpp - Formatting applied via lint.sh Co-authored-by: chhwang <8018170+chhwang@users.noreply.github.com>
21 lines
434 B
C++
21 lines
434 B
C++
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the MIT license.
|
|
|
|
#include <mscclpp/gpu_utils.hpp>
|
|
#include <mscclpp/numa.hpp>
|
|
|
|
#include "../framework.hpp"
|
|
|
|
TEST(NumaTest, Basic) {
|
|
int num;
|
|
MSCCLPP_CUDATHROW(cudaGetDeviceCount(&num));
|
|
if (num == 0) {
|
|
return;
|
|
}
|
|
for (int i = 0; i < num; i++) {
|
|
int numaNode = mscclpp::getDeviceNumaNode(i);
|
|
EXPECT_GE(numaNode, 0);
|
|
mscclpp::numaBind(numaNode);
|
|
}
|
|
}
|