mirror of
https://github.com/QuantumPackage/qp2.git
synced 2024-12-21 19:13:29 +01:00
Accelerated (T)
This commit is contained in:
parent
3aae1dbf77
commit
0fa576f909
@ -287,75 +287,175 @@ subroutine form_w_ijk(nO,nV,i,j,k,T_vvoo,T_ovvo,X_vvvo,X_ovoo,W)
|
|||||||
double precision, intent(out) :: W(nV,nV,nV)!,nO,nO,nO)
|
double precision, intent(out) :: W(nV,nV,nV)!,nO,nO,nO)
|
||||||
|
|
||||||
integer :: l,a,b,c,d
|
integer :: l,a,b,c,d
|
||||||
|
double precision, allocatable, dimension(:,:,:) :: X, Y, Z
|
||||||
|
|
||||||
!W = 0d0
|
!W = 0d0
|
||||||
!do i = 1, nO
|
!do i = 1, nO
|
||||||
! do j = 1, nO
|
! do j = 1, nO
|
||||||
! do k = 1, nO
|
! do k = 1, nO
|
||||||
|
|
||||||
!$OMP PARALLEL &
|
allocate(X(nV,nV,nV))
|
||||||
!$OMP SHARED(nO,nV,i,j,k,T_vvoo,T_ovvo,X_vvvo,X_ovoo,W) &
|
allocate(Y(nV,nV,nV))
|
||||||
!$OMP PRIVATE(a,b,c,d,l) &
|
allocate(Z(nV,nV,nV))
|
||||||
!$OMP DEFAULT(NONE)
|
|
||||||
!$OMP DO collapse(2)
|
|
||||||
do c = 1, nV
|
|
||||||
do b = 1, nV
|
|
||||||
do a = 1, nV
|
|
||||||
W(a,b,c) = 0d0
|
|
||||||
|
|
||||||
do d = 1, nV
|
|
||||||
!W(i,j,k,a,b,c) = W(i,j,k,a,b,c) &
|
|
||||||
W(a,b,c) = W(a,b,c) &
|
|
||||||
! chem (bd|ai)
|
|
||||||
! phys <ba|di>
|
|
||||||
!+ cc_space_v_vvvo(b,a,d,i) * t2(k,j,c,d) &
|
|
||||||
!+ cc_space_v_vvvo(c,a,d,i) * t2(j,k,b,d) & ! bc kj
|
|
||||||
!+ cc_space_v_vvvo(a,c,d,k) * t2(j,i,b,d) & ! prev ac ik
|
|
||||||
!+ cc_space_v_vvvo(b,c,d,k) * t2(i,j,a,d) & ! prev ab ij
|
|
||||||
!+ cc_space_v_vvvo(c,b,d,j) * t2(i,k,a,d) & ! prev bc kj
|
|
||||||
!+ cc_space_v_vvvo(a,b,d,j) * t2(k,i,c,d) ! prev ac ik
|
|
||||||
+ X_vvvo(d,b,a,i) * T_vvoo(d,c,k,j) &
|
|
||||||
+ X_vvvo(d,c,a,i) * T_vvoo(d,b,j,k) & ! bc kj
|
|
||||||
+ X_vvvo(d,a,c,k) * T_vvoo(d,b,j,i) & ! prev ac ik
|
|
||||||
+ X_vvvo(d,b,c,k) * T_vvoo(d,a,i,j) & ! prev ab ij
|
|
||||||
+ X_vvvo(d,c,b,j) * T_vvoo(d,a,i,k) & ! prev bc kj
|
|
||||||
+ X_vvvo(d,a,b,j) * T_vvoo(d,c,k,i) ! prev ac ik
|
|
||||||
enddo
|
|
||||||
|
|
||||||
|
!$OMP PARALLEL DO
|
||||||
|
do b = 1, nV
|
||||||
|
do a = 1, nV
|
||||||
|
do d = 1, nV
|
||||||
|
Z(d,a,b) = X_vvvo(d,b,a,i)
|
||||||
enddo
|
enddo
|
||||||
enddo
|
enddo
|
||||||
enddo
|
enddo
|
||||||
!$OMP END DO nowait
|
!$OMP END PARALLEL DO
|
||||||
|
|
||||||
!$OMP DO collapse(2)
|
call dgemm('T','N',nV*nV,nV,nV, 1.d0, &
|
||||||
|
Z, nV, T_vvoo(1,1,k,j), nV, 0.d0, W, nV*nV)
|
||||||
|
|
||||||
|
!$OMP PARALLEL DO
|
||||||
do c = 1, nV
|
do c = 1, nV
|
||||||
do b = 1, nV
|
do a = 1, nV
|
||||||
do a = 1, nV
|
do d = 1, nV
|
||||||
|
Z(d,a,c) = X_vvvo(d,c,a,i)
|
||||||
do l = 1, nO
|
|
||||||
!W(i,j,k,a,b,c) = W(i,j,k,a,b,c) &
|
|
||||||
W(a,b,c) = W(a,b,c) &
|
|
||||||
! chem (ck|jl)
|
|
||||||
! phys <cj|kl>
|
|
||||||
!- cc_space_v_vooo(c,j,k,l) * t2(i,l,a,b) &
|
|
||||||
!- cc_space_v_vooo(b,k,j,l) * t2(i,l,a,c) & ! bc kj
|
|
||||||
!- cc_space_v_vooo(b,i,j,l) * t2(k,l,c,a) & ! prev ac ik
|
|
||||||
!- cc_space_v_vooo(a,j,i,l) * t2(k,l,c,b) & ! prev ab ij
|
|
||||||
!- cc_space_v_vooo(a,k,i,l) * t2(j,l,b,c) & ! prev bc kj
|
|
||||||
!- cc_space_v_vooo(c,i,k,l) * t2(j,l,b,a) ! prev ac ik
|
|
||||||
- X_ovoo(l,c,j,k) * T_ovvo(l,a,b,i) &
|
|
||||||
- X_ovoo(l,b,k,j) * T_ovvo(l,a,c,i) & ! bc kj
|
|
||||||
- X_ovoo(l,b,i,j) * T_ovvo(l,c,a,k) & ! prev ac ik
|
|
||||||
- X_ovoo(l,a,j,i) * T_ovvo(l,c,b,k) & ! prev ab ij
|
|
||||||
- X_ovoo(l,a,k,i) * T_ovvo(l,b,c,j) & ! prev bc kj
|
|
||||||
- X_ovoo(l,c,i,k) * T_ovvo(l,b,a,j) ! prev ac ik
|
|
||||||
enddo
|
|
||||||
|
|
||||||
enddo
|
enddo
|
||||||
enddo
|
enddo
|
||||||
enddo
|
enddo
|
||||||
!$OMP END DO
|
!$OMP END PARALLEL DO
|
||||||
!$OMP END PARALLEL
|
|
||||||
|
call dgemm('T','N',nV*nV,nV,nV, 1.d0, &
|
||||||
|
Z, nV, T_vvoo(1,1,j,k), nV, 0.d0, Y, nV*nV)
|
||||||
|
|
||||||
|
call dgemm('T','N',nV*nV,nV,nV, 1.d0, &
|
||||||
|
X_vvvo(1,1,1,k), nV, T_vvoo(1,1,j,i), nV, 1.d0, Y, nV*nV)
|
||||||
|
|
||||||
|
call dgemm('T','N',nV,nV*nV,nV, 1.d0, &
|
||||||
|
T_vvoo(1,1,i,j), nV, X_vvvo(1,1,1,k), nV, 1.d0, W, nV)
|
||||||
|
|
||||||
|
call dgemm('T','N',nV,nV*nV,nV, 1.d0, &
|
||||||
|
T_vvoo(1,1,i,k), nV, X_vvvo(1,1,1,j), nV, 1.d0, Y, nV)
|
||||||
|
|
||||||
|
call dgemm('T','N',nV*nV,nV,nV, 1.d0, &
|
||||||
|
X_vvvo(1,1,1,j), nV, T_vvoo(1,1,k,i), nV, 1.d0, W, nV*nV)
|
||||||
|
|
||||||
|
deallocate(Z)
|
||||||
|
|
||||||
|
|
||||||
|
allocate(Z(nO,nV,nV))
|
||||||
|
|
||||||
|
call dgemm('T','N',nV*nV,nV,nO, -1.d0, &
|
||||||
|
T_ovvo(1,1,1,i), nO, X_ovoo(1,1,j,k), nO, 1.d0, W, nV*nV)
|
||||||
|
|
||||||
|
call dgemm('T','N',nV*nV,nV,nO, -1.d0, &
|
||||||
|
T_ovvo(1,1,1,i), nO, X_ovoo(1,1,k,j), nO, 1.d0, Y, nV*nV)
|
||||||
|
|
||||||
|
!$OMP PARALLEL DO
|
||||||
|
do c = 1, nV
|
||||||
|
do a = 1, nV
|
||||||
|
do l = 1, nO
|
||||||
|
Z(l,a,c) = T_ovvo(l,c,a,k)
|
||||||
|
enddo
|
||||||
|
enddo
|
||||||
|
enddo
|
||||||
|
!$OMP END PARALLEL DO
|
||||||
|
|
||||||
|
call dgemm('T','N',nV*nV,nV,nO, -1.d0, &
|
||||||
|
Z, nO, X_ovoo(1,1,i,j), nO, 1.d0, Y, nV*nV)
|
||||||
|
|
||||||
|
call dgemm('T','N',nV,nV*nV,nO, -1.d0, &
|
||||||
|
X_ovoo(1,1,j,i), nO, T_ovvo(1,1,1,k), nO, 1.d0, Y, nV)
|
||||||
|
|
||||||
|
call dgemm('T','N',nV,nV*nV,nO, -1.d0, &
|
||||||
|
X_ovoo(1,1,k,i), nO, T_ovvo(1,1,1,j), nO, 1.d0, W, nV)
|
||||||
|
|
||||||
|
!$OMP PARALLEL DO
|
||||||
|
do b = 1, nV
|
||||||
|
do a = 1, nV
|
||||||
|
do l = 1, nO
|
||||||
|
Z(l,a,b) = T_ovvo(l,b,a,j)
|
||||||
|
enddo
|
||||||
|
enddo
|
||||||
|
enddo
|
||||||
|
!$OMP END PARALLEL DO
|
||||||
|
|
||||||
|
call dgemm('T','N',nV*nV,nV,nO, -1.d0, &
|
||||||
|
Z, nO, X_ovoo(1,1,i,k), nO, 1.d0, W, nV*nV)
|
||||||
|
|
||||||
|
!$OMP PARALLEL DO
|
||||||
|
do c = 1, nV
|
||||||
|
do b = 1, nV
|
||||||
|
do a = 1, nV
|
||||||
|
W(a,b,c) = W(a,b,c) + Y(a,c,b)
|
||||||
|
enddo
|
||||||
|
enddo
|
||||||
|
enddo
|
||||||
|
!$OMP END PARALLEL DO
|
||||||
|
|
||||||
|
deallocate(X,Y,Z)
|
||||||
|
|
||||||
|
|
||||||
|
! !$OMP PARALLEL &
|
||||||
|
! !$OMP SHARED(nO,nV,i,j,k,T_vvoo,T_ovvo,X_vvvo,X_ovoo,W) &
|
||||||
|
! !$OMP PRIVATE(a,b,c,d,l) &
|
||||||
|
! !$OMP DEFAULT(NONE)
|
||||||
|
!
|
||||||
|
! !$OMP DO collapse(2)
|
||||||
|
! do c = 1, nV
|
||||||
|
! do b = 1, nV
|
||||||
|
! do a = 1, nV
|
||||||
|
! W(a,b,c) = 0.d0
|
||||||
|
!
|
||||||
|
! do d = 1, nV
|
||||||
|
! !W(i,j,k,a,b,c) = W(i,j,k,a,b,c) &
|
||||||
|
! W(a,b,c) = W(a,b,c) &
|
||||||
|
! ! chem (bd|ai)
|
||||||
|
! ! phys <ba|di>
|
||||||
|
! !+ cc_space_v_vvvo(b,a,d,i) * t2(k,j,c,d) &
|
||||||
|
! !+ cc_space_v_vvvo(c,a,d,i) * t2(j,k,b,d) & ! bc kj
|
||||||
|
! !+ cc_space_v_vvvo(a,c,d,k) * t2(j,i,b,d) & ! prev ac ik
|
||||||
|
! !+ cc_space_v_vvvo(b,c,d,k) * t2(i,j,a,d) & ! prev ab ij
|
||||||
|
! !+ cc_space_v_vvvo(c,b,d,j) * t2(i,k,a,d) & ! prev bc kj
|
||||||
|
! !+ cc_space_v_vvvo(a,b,d,j) * t2(k,i,c,d) ! prev ac ik
|
||||||
|
! + X_vvvo(d,b,a,i) * T_vvoo(d,c,k,j) &
|
||||||
|
! + X_vvvo(d,c,a,i) * T_vvoo(d,b,j,k) & ! bc kj
|
||||||
|
! + X_vvvo(d,a,c,k) * T_vvoo(d,b,j,i) & ! prev ac ik
|
||||||
|
! + X_vvvo(d,b,c,k) * T_vvoo(d,a,i,j) & ! prev ab ij
|
||||||
|
! + X_vvvo(d,c,b,j) * T_vvoo(d,a,i,k) & ! prev bc kj
|
||||||
|
! + X_vvvo(d,a,b,j) * T_vvoo(d,c,k,i) ! prev ac ik
|
||||||
|
! enddo
|
||||||
|
!
|
||||||
|
! enddo
|
||||||
|
! enddo
|
||||||
|
! enddo
|
||||||
|
! !$OMP END DO nowait
|
||||||
|
!
|
||||||
|
! !$OMP DO collapse(2)
|
||||||
|
! do c = 1, nV
|
||||||
|
! do b = 1, nV
|
||||||
|
! do a = 1, nV
|
||||||
|
!
|
||||||
|
! do l = 1, nO
|
||||||
|
! !W(i,j,k,a,b,c) = W(i,j,k,a,b,c) &
|
||||||
|
! W(a,b,c) = W(a,b,c) &
|
||||||
|
! ! chem (ck|jl)
|
||||||
|
! ! phys <cj|kl>
|
||||||
|
! !- cc_space_v_vooo(c,j,k,l) * t2(i,l,a,b) &
|
||||||
|
! !- cc_space_v_vooo(b,k,j,l) * t2(i,l,a,c) & ! bc kj
|
||||||
|
! !- cc_space_v_vooo(b,i,j,l) * t2(k,l,c,a) & ! prev ac ik
|
||||||
|
! !- cc_space_v_vooo(a,j,i,l) * t2(k,l,c,b) & ! prev ab ij
|
||||||
|
! !- cc_space_v_vooo(a,k,i,l) * t2(j,l,b,c) & ! prev bc kj
|
||||||
|
! !- cc_space_v_vooo(c,i,k,l) * t2(j,l,b,a) ! prev ac ik
|
||||||
|
! - T_ovvo(l,a,b,i) * X_ovoo(l,c,j,k) &
|
||||||
|
! - T_ovvo(l,a,c,i) * X_ovoo(l,b,k,j) & ! bc kj
|
||||||
|
! - T_ovvo(l,c,a,k) * X_ovoo(l,b,i,j) & ! prev ac ik
|
||||||
|
! - T_ovvo(l,c,b,k) * X_ovoo(l,a,j,i) & ! prev ab ij
|
||||||
|
! - T_ovvo(l,b,c,j) * X_ovoo(l,a,k,i) & ! prev bc kj
|
||||||
|
! - T_ovvo(l,b,a,j) * X_ovoo(l,c,i,k) ! prev ac ik
|
||||||
|
! enddo
|
||||||
|
!
|
||||||
|
! enddo
|
||||||
|
! enddo
|
||||||
|
! enddo
|
||||||
|
! !$OMP END DO
|
||||||
|
! !$OMP END PARALLEL
|
||||||
|
|
||||||
! enddo
|
! enddo
|
||||||
! enddo
|
! enddo
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user