f1275f52c1
* Rewrite gpu_id related code. * Remove normalised/unnormalised operatios. * Address difference between `Index' and `Device ID'. * Modify doc for `gpu_id'. * Better LOG for GPUSet. * Check specified n_gpus. * Remove inappropriate `device_idx' term. * Clarify GpuIdType and size_t.
38 lines
925 B
C++
38 lines
925 B
C++
#include "../../../src/common/common.h"
|
|
#include <gtest/gtest.h>
|
|
|
|
namespace xgboost {
|
|
TEST(GPUSet, Basic) {
|
|
GPUSet devices = GPUSet::Empty();
|
|
ASSERT_TRUE(devices.IsEmpty());
|
|
|
|
devices = GPUSet{0, 1};
|
|
ASSERT_TRUE(devices != GPUSet::Empty());
|
|
EXPECT_EQ(devices.Size(), 1);
|
|
|
|
devices = GPUSet::Range(1, 0);
|
|
EXPECT_EQ(devices.Size(), 0);
|
|
EXPECT_TRUE(devices.IsEmpty());
|
|
|
|
EXPECT_FALSE(devices.Contains(1));
|
|
|
|
devices = GPUSet::Range(2, -1);
|
|
EXPECT_EQ(devices, GPUSet::Empty());
|
|
EXPECT_EQ(devices.Size(), 0);
|
|
EXPECT_TRUE(devices.IsEmpty());
|
|
|
|
devices = GPUSet::Range(2, 8); // 2 ~ 10
|
|
EXPECT_EQ(devices.Size(), 8);
|
|
EXPECT_ANY_THROW(devices.DeviceId(8));
|
|
|
|
auto device_id = devices.DeviceId(0);
|
|
EXPECT_EQ(device_id, 2);
|
|
auto device_index = devices.Index(2);
|
|
EXPECT_EQ(device_index, 0);
|
|
|
|
#ifndef XGBOOST_USE_CUDA
|
|
EXPECT_EQ(GPUSet::AllVisible(), GPUSet::Empty());
|
|
#endif
|
|
}
|
|
} // namespace xgboost
|