Function TEST

test/singa/test_platform.cc:28–38

using singa::Platform;

TEST(Platform, CreateMultDevice) {
  int n = Platform::GetNumGPUs();
  auto devs = Platform::CreateCudaGPUs(n);
  for (size_t i = 0; i < devs.size(); i++) {
    auto b = devs[i]->NewBlock(512 + 512 * (2 - i));
    // for lazy allocation
    b->mutable_data();
    EXPECT_EQ(512 + 512 * (2 - i), devs[i]->GetAllocatedMem());
    devs[i]->FreeBlock(b);
  }
}

TEST(Platform, NumGPUs) {
  int n = Platform::GetNumGPUs();