24小时热门版块排行榜    

查看: 540  |  回复: 0

panda0927

木虫 (初入文坛)

[交流] 并行的一维fft fortran代码, 速度还行

subroutine cfftserial ( n, x )
   implicit none
   integer (kind = 4) :: n, i
   real (kind = 8)     :: pi, pid
   complex (kind = 8) :: x(n, 2)
   
   pi = acos ( -1.0d0 )
   pid = -pi * 2.0d0
   pid = pid / dble ( n )
!$omp parallel do default(shared) private(i)   
   do i = 0, n-1
      x(i+1, 1) = cmplx ( cos ( pid * dble ( i ) ), sin ( pid * dble ( i ) ), kind = 8 )
      x(i+1, 2) = conjg (  x(i+1, 1) )
   end do
!$omp end parallel do  

   return
end subroutine cfftserial

subroutine ifftserial ( n, x )
   implicit none
   integer (kind = 4) :: n, i, j, k, m, it, is
   integer (kind = 4) :: x(n)
   
   k = 0
   m = 1
   do  
      m = m * 2
      k = k + 1
      if ( n <= m ) exit
   end do
   
!$omp parallel do default(shared) private(it, m, is, i, j)   
   do it = 0, n - 1
      m = it
      is = 0
      do i = 0, k - 1
         j = m / 2
         is = 2 * is + (m - 2 * j)
         m = j
      end do
      x(it + 1) = is + 1
   end do
!$omp end parallel do
   
   return
end subroutine ifftserial

subroutine purefour(n, isign, iserial, cserial, fft)
   implicit none
   integer (kind = 4) :: n, it, is, m, k, i, j, nv, l0, isign
   integer (kind = 4) :: iserial(n)
   real (kind = 8)     :: pi , pid, dn
   complex (kind = 8) :: temp1, temp2, xsi
   complex (kind = 8) :: fft(n), temp(n), cserial(n, 2)
   
   dn = 1.0d0 / dble ( n )
   k = 0
   m = 1
   do  
      m = m * 2
      k = k + 1
      if ( n <= m ) exit
   end do

!$omp parallel default(shared)   

!$omp do private(i)   
   do i = 1, n
      temp(i) = fft(iserial(i))
   end do
!$omp end do

!$omp do private(j, temp2)
   do j = 0, n - 2, 2
      temp2 = temp(j + 1)
      temp(j + 1) = temp2 + temp(j + 2)  
      temp(j + 2) = temp2 - temp(j + 2)
   end do
!$omp end do

!$omp end parallel
   
   if ( isign == 1) then
      is = 1
   else
      is = 2
   end if

   m = n / 2
   nv = 2

   do l0 = k - 2, 0, -1
      m = m / 2
      nv = nv * 2
!$omp parallel do  &
!$omp default(private) &
!$omp shared(m, nv, temp, cserial, is)
      do it = 0, (m - 1) * nv, nv
         do j = 0, (nv / 2) - 1
            i = it + j + 1
            temp2 = cserial(m * j + 1, is) * temp(i + nv / 2)
            temp(i + nv / 2) = temp(i) - temp2
            temp(i) = temp(i) + temp2
         end do
      end do
!$omp end parallel do
   end do
   
   if ( isign == -1 ) then
!$omp parallel do &
!$omp default(shared)  private(i)
      do i = 1, n
         fft(i) = temp(i) * dn
      end do
!$omp end parallel do
   else
!$omp parallel do &
!$omp default(shared)  private(i)
      do i = 1, n
         fft(i) = temp(i)
      end do
!$omp end parallel do
   end if

   return
end subroutine purefour
回复此楼
已阅   回复此楼   关注TA 给TA发消息 送TA红花 TA的回帖

智能机器人

Robot (super robot)

我们都爱小木虫

相关版块跳转 我要订阅楼主 panda0927 的主题更新
普通表情 龙 兔 虎 猫 高级回复 (可上传附件)
最具人气热帖推荐 [查看全部] 作者 回/看 最后发表
[考博] 售SCI文章,我:8O.5.5.1O.54,科目全,可十急 +4 5sJFDKp2kwuI 2026-11-06 11/550 2026-11-09 02:48 by l48hFd8Nwae2
[考研] 售SCI一区T0P文章,我:8O.55.1.O.54,科目全,可伽急 +3 5sJFDKp2kwuI 2026-11-06 9/450 2026-11-09 02:41 by l48hFd8Nwae2
[考研] 售SCI文章,我:8O.5.5.1O.54,科目全,可十急 +5 5sJFDKp2kwuI 2026-11-06 9/450 2026-11-09 01:42 by l48hFd8Nwae2
[论文投稿] 售SCI一区文章,我:8.O.55.1.O.54,科目齐全,可伽急 +3 5sJFDKp2kwuI 2026-11-06 9/450 2026-11-09 01:35 by l48hFd8Nwae2
[硕博家园] 售SCI一区T0P文章,我:8.O55.1.O.54,科目全,可十急 +5 2rKfVQg2jQww 2026-11-06 11/550 2026-11-09 00:26 by l48hFd8Nwae2
[公派出国] 售SCI一区文章,我:8.O.55.1.O.54,科目齐全,可伽急 +4 5sJFDKp2kwuI 2026-11-06 8/400 2026-11-09 00:14 by l48hFd8Nwae2
[教师之家] 售SCI一区T0P文章,我:8.O.55.1.O.5.4,科目全,可+急 +5 5sJFDKp2kwuI 2026-11-06 9/450 2026-11-08 22:46 by l48hFd8Nwae2
[硕博家园] 售SCI一区文章,我:8O5.5.1.O5.4,科目全,可伽急 +3 5sJFDKp2kwuI 2026-11-06 9/450 2026-11-08 21:29 by l48hFd8Nwae2
[考博] 售SCI一区文章,我:8.O.55.1.O.54,科目齐全,可伽急 +10 ncTVBM6AOJHF 2026-11-04 22/1100 2026-11-06 17:18 by 5sJFDKp2kwuI
[考研] 售SCI一区T0P文章,我:8.O55.1.O.54,科目全,可十急 +4 ELBZ6WwEaRj6 2026-11-05 7/350 2026-11-06 16:43 by 5sJFDKp2kwuI
[公派出国] 售一区SCI文章T0P,我:8O.551.O54,科目全,可十急 +4 ELBZ6WwEaRj6 2026-11-05 5/250 2026-11-06 16:43 by 5sJFDKp2kwuI
[硕博家园] 售SCI一区文章,我:8O5.5.1.O5.4,科目全,可伽急 +4 ttH3f1oA6ace 2026-11-04 11/550 2026-11-06 16:10 by 5sJFDKp2kwuI
[公派出国] 售SCI一区T0P文章,我:8O.55.1.O.5.4,科目齐全,可+急 +6 ncTVBM6AOJHF 2026-11-04 15/750 2026-11-06 13:49 by BkA9Bi5cMhkD
[考博] 售SCI一区T0P文章,我:8.O55.1.O.54,科目全,可十急 +7 ncTVBM6AOJHF 2026-11-04 17/850 2026-11-06 13:47 by BkA9Bi5cMhkD
[考研] 售一区SCI文章T0P,我:8O.551.O54,科目全,可十急 +8 ncTVBM6AOJHF 2026-11-04 18/900 2026-11-06 13:36 by BkA9Bi5cMhkD
[考研] 售SCI一区T0P文章,我:8.O.55.1.O.54,科目齐全,可+急 +5 ELBZ6WwEaRj6 2026-11-05 10/500 2026-11-06 12:32 by BkA9Bi5cMhkD
[硕博家园] 售一区SCI文章T0P,我:8O.551.O54,科目全,可十急 +3 ELBZ6WwEaRj6 2026-11-05 5/250 2026-11-06 09:01 by 2lRNluKrJpwL
[考博] 售SCI一区T0P文章,我:8.O.55.1.O54,科目全,可伽急 +4 kvewwGtKktqB 2026-11-04 11/550 2026-11-06 05:10 by qL1cU46YIdK0
[考研] 售SCI一区T0P文章,我:8.O55.1.O.54,科目全,可十急 +4 ttH3f1oA6ace 2026-11-04 10/500 2026-11-06 02:04 by qL1cU46YIdK0
[考博] 售SCI一区T0P文章,我:8.O.55.1.O.5.4,科目全,可+急 +6 ncTVBM6AOJHF 2026-11-04 10/500 2026-11-06 01:51 by qL1cU46YIdK0
信息提示
请填处理意见