mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-13 09:46:00 +00:00
8d1b984bed
* Changed device handle interfaces * Changed proxy service interfaces * Move device code into separate files * Fixed FIFO polling issues * Add configuration arguments in several interface functions --------- Co-authored-by: Changho Hwang <changhohwang@microsoft.com> Co-authored-by: Binyang Li <binyli@microsoft.com> Co-authored-by: root <root@a100-saemal0.qxveptpukjsuthqvv514inp03c.gx.internal.cloudapp.net>
21 lines
432 B
C++
21 lines
432 B
C++
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the MIT license.
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <mscclpp/cuda_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);
|
|
}
|
|
}
|