C++ (Cpp) setcudavisibledevicesの例

コード例 #1

ファイル: test_uut.c プロジェクト: brianrtc/torque

END_TEST


START_TEST(test_setcudavisibledevices)
  {
  char  curr[3];
  char  *cp;
  std::stringstream output;

  setcudavisibledevices("1");
  fail_unless(MOMCudaVisibleDevices == 1, "did not set cuda_visble_devices to 1");

  setcudavisibledevices("0");
  fail_unless(MOMCudaVisibleDevices == 0, "did not set cuda_visble_devices to 0");

  strcpy(curr, " 1");
  cp = curr;
  set_report_mom_cuda_visible_devices(output, cp);
  std::string result = output.str();
  fail_unless(strcmp(result.c_str(), "cuda_visible_devices=1"));

  strcpy(curr, " 0");
  cp = curr;
  set_report_mom_cuda_visible_devices(output, cp);
  result = output.str();
  fail_unless(strcmp(result.c_str(), "cuda_visible_devices=0"));

  }

コード例 #2

ファイル: test_uut.c プロジェクト: dooodlesomething/torque

END_TEST


START_TEST(test_setcudavisibledevices)
  {
  char  curr[3];
  char  *cp;
  std::stringstream output;

  setcudavisibledevices("1");
  fail_unless(MOMCudaVisibleDevices == 1, "did not set cuda_visble_devices to 1");

  setcudavisibledevices("0");
  fail_unless(MOMCudaVisibleDevices == 0, "did not set cuda_visble_devices to 0");
  }