Added a function for loading ScalarArrays to the GPU

2019-09-05 16:35:08 +03:00
parent cdb504e772
commit 263a1d23a3
2 changed files with 15 additions and 0 deletions
--- a/src/core/device.cu
+++ b/src/core/device.cu
@@ -447,6 +447,16 @@ acDeviceLoadInt3Constant(const Device device, const Stream stream, const AcInt3P
    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
 acDeviceLoadMeshInfo(const Device device, const Stream stream, const AcMeshInfo device_config)
 {