sunwen/Aurora
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix getvalue

Browse Source
This commit is contained in:
kradchen
2023-11-27 10:18:55 +08:00
parent 77e1ea23c0
commit 7d9d99e5a2
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/CudaMatrix.cpp
View File

@@ -1062,7 +1062,7 @@ bool CudaMatrix::setBlock(int aDim,int aBeginIndex, int aEndIndex, const CudaMat
unaryCompare(this->getData(),aMatrix.getData(),data,this->getDataSize(),::NE); unaryCompare(this->getData(),aMatrix.getData(),data,this->getDataSize(),::NE);
return out; return out;
} }
float CudaMatrix::getValue(size_t index){ float CudaMatrix::getValue(size_t index) const{
float result; float result;
cudaError_t cuda_error = cudaMemcpy(&result, getData() + index, sizeof(float), cudaMemcpyDeviceToHost); cudaError_t cuda_error = cudaMemcpy(&result, getData() + index, sizeof(float), cudaMemcpyDeviceToHost);

2
src/CudaMatrix.h
View File

@@ -113,7 +113,7 @@ namespace Aurora
CudaMatrix operator!=(const CudaMatrix &aMatrix) const; CudaMatrix operator!=(const CudaMatrix &aMatrix) const;
// sub // sub
float getValue(size_t index); float getValue(size_t index) const;
void setValue(size_t index, const float& value); void setValue(size_t index, const float& value);
/** /**