mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-13 17:55:52 +00:00
21 lines
431 B
C++
21 lines
431 B
C++
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the MIT license.
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <mscclpp/gpu_utils.hpp>
|
|
#include <mscclpp/numa.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);
|
|
}
|
|
}
|