17 lines
377 B
Plaintext
17 lines
377 B
Plaintext
|
#include <cuda/std/memory>
|
||
|
|
|
||
|
|
#include <cuda_runtime_api.h>
|
||
|
|
|
||
|
|
#include <unittest/unittest.h>
|
||
|
|
|
||
|
|
template <typename T>
|
||
|
|
void TestCudaMallocResultAligned(const std::size_t n)
|
||
|
|
{
|
||
|
|
T* ptr = nullptr;
|
||
|
|
cudaMalloc(&ptr, n * sizeof(T));
|
||
|
|
cudaFree(ptr);
|
||
|
|
|
||
|
|
ASSERT_EQUAL(true, ::cuda::std::is_sufficiently_aligned<alignof(T)>(ptr));
|
||
|
|
}
|
||
|
|
DECLARE_VARIABLE_UNITTEST(TestCudaMallocResultAligned);
|