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

Fix abs bug

Browse Source
This commit is contained in:
Krad
2023-04-25 17:37:22 +08:00
parent cf96572074
commit c71e359ce5
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/Function1D.cpp
View File

@@ -128,7 +128,7 @@ Aurora::Matrix Aurora::abs(const Aurora::Matrix &matrix) {
else{ else{
vzAbsI(matrix.getDataSize(), (std::complex<double> *)matrix.getData(), SAME_STRIDE,output, SAME_STRIDE); vzAbsI(matrix.getDataSize(), (std::complex<double> *)matrix.getData(), SAME_STRIDE,output, SAME_STRIDE);
} }
return Aurora::Matrix::New(output, matrix); return Aurora::Matrix::New(output, matrix.getDimSize(0), matrix.getDimSize(1), matrix.getDimSize(2));
} }
Aurora::Matrix Aurora::abs(Aurora::Matrix&& matrix) { Aurora::Matrix Aurora::abs(Aurora::Matrix&& matrix) {
@@ -140,7 +140,7 @@ Aurora::Matrix Aurora::abs(Aurora::Matrix&& matrix) {
else{ else{
auto output = malloc(matrix.getDataSize()); auto output = malloc(matrix.getDataSize());
vzAbsI(matrix.getDataSize(), (std::complex<double> *)matrix.getData(), SAME_STRIDE,output, SAME_STRIDE); vzAbsI(matrix.getDataSize(), (std::complex<double> *)matrix.getData(), SAME_STRIDE,output, SAME_STRIDE);
return Aurora::Matrix::New(output, matrix); return Aurora::Matrix::New(output, matrix.getDimSize(0), matrix.getDimSize(1), matrix.getDimSize(2));
} }
} }