@ -16,8 +16,8 @@
#include "paddle/memory/detail/system_allocator.h"
#include <vector>
#include <mutex>
namespace paddle {
namespace memory {
@ -30,8 +30,7 @@ inline void throw_on_error(cudaError_t e, const char* message) {
int GetDeviceCount(void) {
int count;
throw_on_error(cudaGetDeviceCount(&count),
"cudaGetDeviceCount failed");
throw_on_error(cudaGetDeviceCount(&count), "cudaGetDeviceCount failed");
return count;
}