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

Fix zcopy.

Browse Source
This commit is contained in:
sunwen
2023-10-09 09:46:24 +08:00
parent 8167775d38
commit f8f9e453b5
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/Function2D.cpp
View File

@@ -714,7 +714,7 @@ Matrix Aurora::fft(const Matrix &aMatrix, long aFFTSize) {
} else {
//按列copy原值
for (int i = 0 ; i < aMatrix.getDimSize(1); ++i) {
cblas_zcopy(needCopySize, aMatrix.getData()+i*aMatrix.getDimSize(0)*2, 1, output+i*rowSize*2, 1);
cblas_ccopy(needCopySize, aMatrix.getData()+i*aMatrix.getDimSize(0)*2, 1, output+i*rowSize*2, 1);
}
}
@@ -787,7 +787,7 @@ Matrix Aurora::ifft(const Matrix &aMatrix, long aFFTSize ) {
} else {
//按列copy原值
for (int i = 0 ; i < aMatrix.getDimSize(1); ++i) {
cblas_zcopy(needCopySize, aMatrix.getData()+i*aMatrix.getDimSize(0)*2, 1, output+i*rowSize*2, 1);
cblas_ccopy(needCopySize, aMatrix.getData()+i*aMatrix.getDimSize(0)*2, 1, output+i*rowSize*2, 1);
}
}
//创建 Descriptor, 精度 float , 输入类型实数, 维度1
@@ -850,7 +850,7 @@ Matrix Aurora::ifft_symmetric(const Matrix &aMatrix,long length)
//所有数据统一置0
cblas_scopy(length*2,&zero,0,calcData,1);
//copy前半段数据
cblas_zcopy(copyLength,aMatrix.getData(),1,calcData,1);
cblas_ccopy(copyLength,aMatrix.getData(),1,calcData,1);
//copy后半段数据,跳过index 0的值,并设置虚部共轭
vsAddI(copyLength-1,&zero,0,(aMatrix.getData()+2),2,(calcData+(length-1)*2),-2);
vsSubI(copyLength-1,&zero,0,(aMatrix.getData()+2+1),2,(calcData+(length-1)*2+1),-2);

12
src/Matrix.cpp
View File

@@ -1492,7 +1492,7 @@ namespace Aurora {
// cblas_zcopy_batch_strided(mSize,(std::complex<float>*)slice.mData,slice.mStride,slice.mStride2,
// (std::complex<float>*)mData,mStride,mStride2,mSize2);
for (int i = 0; i < mSize2; ++i) {
cblas_zcopy(mSize, (std::complex<float> *) (slice.mData + i * slice.mStride2), slice.mStride,
cblas_ccopy(mSize, (std::complex<float> *) (slice.mData + i * slice.mStride2), slice.mStride,
(std::complex<float> *) (mData + i * mStride2), mStride);
}
}
@@ -1503,7 +1503,7 @@ namespace Aurora {
cblas_scopy(mSize,slice.mData,slice.mStride,mData,mStride);
}
else {
cblas_zcopy(mSize, (std::complex<float> *)slice.mData, slice.mStride, (std::complex<float> *)mData, mStride);
cblas_ccopy(mSize, (std::complex<float> *)slice.mData, slice.mStride, (std::complex<float> *)mData, mStride);
}
break;
}
@@ -1575,7 +1575,7 @@ namespace Aurora {
mStride2, mSize2);
}
else {
cblas_zcopy_batch_strided(mSize,(std::complex<float>*)matrix.getData(),1,matrix.getDimSize(0),
cblas_ccopy_batch_strided(mSize,(std::complex<float>*)matrix.getData(),1,matrix.getDimSize(0),
(std::complex<float>*)mData,mStride,mStride2,mSize2);
}
break;
@@ -1586,7 +1586,7 @@ namespace Aurora {
cblas_scopy(mSize,matrix.getData(),1,mData,mStride);
}
else {
cblas_zcopy(mSize, (std::complex<float> *) matrix.getData(),1,
cblas_ccopy(mSize, (std::complex<float> *) matrix.getData(),1,
(std::complex<float> *) mData, mStride);
}
break;
@@ -1654,7 +1654,7 @@ namespace Aurora {
mStride2,data, 1, mSize, mSize2);
}
else {
cblas_zcopy_batch_strided(mSize, (std::complex<float> *) mData, mStride, mStride2,
cblas_ccopy_batch_strided(mSize, (std::complex<float> *) mData, mStride, mStride2,
(std::complex<float> *) data, 1, mSize,
mSize2);
}
@@ -1665,7 +1665,7 @@ namespace Aurora {
cblas_scopy(mSize,mData,mStride,data,1);
}
else {
cblas_zcopy(mSize, (std::complex<float> *) mData, mStride,
cblas_ccopy(mSize, (std::complex<float> *) mData, mStride,
(std::complex<float> *) data, 1);
}
break;