ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
|
#include <vector>
#include <gtest/gtest.h>
#include <ATen/core/tensor.h>
#include <base/core/allocator.h>
#include <base/core/cpu_allocator.h>
Functions | |
TEST (CPUAllocator, AllocateAndFree) | |
TEST (CPUAllocator, AllocatedSize) | |
TEST (CPUAllocator, GetDeviceType) | |
TEST | ( | CPUAllocator | , |
AllocateAndFree | |||
) |
TEST | ( | CPUAllocator | , |
AllocatedSize | |||
) |
TEST | ( | CPUAllocator | , |
GetDeviceType | |||
) |