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")); }
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"); }