提交 6250d108 编写于 作者: L liaogang

FIX: clang-format

上级 ce70df86
......@@ -16,8 +16,8 @@ limitations under the License. */
#include <stddef.h> // for size_t
#include <thrust/system_error.h>
#include <thrust/system/cuda/error.h>
#include <thrust/system_error.h>
namespace paddle {
namespace memory {
......@@ -36,14 +36,14 @@ inline void throw_on_error(cudaError_t e, const char* message) {
// default, we should use GPUAllocator<staging=false>.
template <bool staging>
class GPUAllocator {
public:
public:
void* Alloc(size_t size);
void Free(void* p, size_t size);
};
template <>
class GPUAllocator<false> {
public:
public:
void* Alloc(size_t size) {
void* p = 0;
cudaError_t result = cudaMalloc(&p, size);
......@@ -70,7 +70,7 @@ public:
template <>
class GPUAllocator<true> {
public:
public:
void* Alloc(size_t size) {
void* p = 0;
cudaError_t result = cudaMallocHost(&p, size);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册