cwpearson/astaroth
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added a function for loading ScalarArrays to the GPU

Browse Source
This commit is contained in:
jpekkila
2019-09-05 16:35:08 +03:00
parent cdb504e772
commit 263a1d23a3
2 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
include/astaroth_device.h
View File

@@ -61,6 +61,11 @@ AcResult acDeviceLoadIntConstant(const Device device, const Stream stream, const
AcResult acDeviceLoadInt3Constant(const Device device, const Stream stream, const AcInt3Param param, AcResult acDeviceLoadInt3Constant(const Device device, const Stream stream, const AcInt3Param param,
const int3 value); const int3 value);
/** */
AcResult acDeviceLoadScalarArray(const Device device, const Stream stream,
const ScalarArrayHandle handle, const AcReal* data,
const size_t num);
/** */ /** */
AcResult acDeviceLoadMeshInfo(const Device device, const Stream stream, AcResult acDeviceLoadMeshInfo(const Device device, const Stream stream,
const AcMeshInfo device_config); const AcMeshInfo device_config);

10
src/core/device.cu
View File

@@ -447,6 +447,16 @@ acDeviceLoadInt3Constant(const Device device, const Stream stream, const AcInt3P
return AC_SUCCESS; return AC_SUCCESS;
} }
AcResult
acDeviceLoadScalarArray(const Device device, const Stream stream, const ScalarArrayHandle handle,
const AcReal* data, const size_t num)
{
cudaSetDevice(device->id);
ERRCHK_CUDA(cudaMemcpyAsync(device->vba.profiles[handle], data, sizeof(data[0]) * num,
cudaMemcpyHostToDevice, device->streams[stream]));
return AC_SUCCESS;
}
AcResult AcResult
acDeviceLoadMeshInfo(const Device device, const Stream stream, const AcMeshInfo device_config) acDeviceLoadMeshInfo(const Device device, const Stream stream, const AcMeshInfo device_config)
{ {