63 lines
1.3 KiB
Plaintext
63 lines
1.3 KiB
Plaintext
|
//
|
||
|
|
// Created by kai on 02/02/23.
|
||
|
|
//
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#ifndef VERLET_CUDAMEMORY_CUH
|
||
|
|
#define VERLET_CUDAMEMORY_CUH
|
||
|
|
|
||
|
|
#include "definitions.cuh"
|
||
|
|
#include <cstdio>
|
||
|
|
|
||
|
|
|
||
|
|
template<typename T>
|
||
|
|
class CudaMemory {
|
||
|
|
public:
|
||
|
|
explicit CudaMemory(size_t array_length) {
|
||
|
|
CUDA_CALL(cudaMalloc(&this->devPtr, sizeof(T) * array_length));
|
||
|
|
this->hostPtr = (T*)(malloc(sizeof(T) * array_length));
|
||
|
|
memset(this->hostPtr, 0, array_length);
|
||
|
|
this->N = array_length;
|
||
|
|
}
|
||
|
|
|
||
|
|
CudaMemory() {
|
||
|
|
CUDA_CALL(cudaMallocManaged(&this->devPtr, sizeof(T)));
|
||
|
|
this->hostPtr = (T*)malloc(sizeof(T));
|
||
|
|
memset(this->hostPtr, 0, 1);
|
||
|
|
this->N = 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
~CudaMemory() {
|
||
|
|
CUDA_CALL(cudaFree(this->devPtr))
|
||
|
|
free(this->hostPtr);
|
||
|
|
}
|
||
|
|
|
||
|
|
T* getPointer() {
|
||
|
|
return this->hostPtr;
|
||
|
|
}
|
||
|
|
|
||
|
|
T* getDevicePointer() {
|
||
|
|
return this->devPtr;
|
||
|
|
}
|
||
|
|
|
||
|
|
void send() {
|
||
|
|
CUDA_CALL(cudaMemcpy(this->devPtr, this->hostPtr, sizeof(T) * this->N, cudaMemcpyHostToDevice));
|
||
|
|
}
|
||
|
|
|
||
|
|
void sync() {
|
||
|
|
CUDA_CALL(cudaMemcpy(this->hostPtr, this->devPtr, sizeof(T) * this->N, cudaMemcpyDeviceToHost));
|
||
|
|
}
|
||
|
|
|
||
|
|
T& operator[](size_t idx) {
|
||
|
|
return this->hostPtr[idx];
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
T* hostPtr;
|
||
|
|
T* devPtr;
|
||
|
|
size_t N;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif //VERLET_CUDAMEMORY_CUH
|