Merge branch 'dtof' of http://192.168.1.9:3000/Bug/Aurora into dtof

2023-11-27 10:19:52 +08:00
parent 7d9d99e5a2 edff6018bb
commit 980865b494
4 changed files with 161 additions and 2 deletions
--- a/src/CudaMatrix.cpp
+++ b/src/CudaMatrix.cpp
@@ -147,6 +147,10 @@ CudaMatrix CudaMatrix::deepCopy() const

 Matrix CudaMatrix::toHostMatrix() const
 {
+    if(!mData.get())
+    {
+        return Matrix();
+    }
    unsigned long long size = getDataSize() * getValueType();
    float* data = new float[size];
    cudaMemcpy(data, mData.get(), sizeof(float) * size, cudaMemcpyDeviceToHost);