From b467bef6dd1e14c5914cc6508aa898d5f1665e3a Mon Sep 17 00:00:00 2001
From: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
Date: Fri, 28 Jun 2024 21:37:14 +0200
Subject: [PATCH] Forgot file

---
 src/ccsd/ccsd_space_orb_sub.irp.f | 68 +++++++++++++++----------------
 src/gpu/gpu.irp.f                 | 11 +++++
 2 files changed, 45 insertions(+), 34 deletions(-)
 create mode 100644 src/gpu/gpu.irp.f

diff --git a/src/ccsd/ccsd_space_orb_sub.irp.f b/src/ccsd/ccsd_space_orb_sub.irp.f
index 4e06e31d..5c2daa05 100644
--- a/src/ccsd/ccsd_space_orb_sub.irp.f
+++ b/src/ccsd/ccsd_space_orb_sub.irp.f
@@ -271,40 +271,40 @@ subroutine ccsd_energy_space_x(nO,nV,d_cc_space_v_oovv,d_cc_space_f_vo,tau_x,t1,
   integer :: i,j,a,b
   double precision :: e
 
-  energy = 0d0
-  !$omp parallel &
-  !$omp shared(nO,nV,energy,tau_x,t1,&
-  !$omp d_cc_space_f_vo,d_cc_space_v_oovv) &
-  !$omp private(i,j,a,b,e) &
-  !$omp default(none)
-  e = 0d0
-  !$omp do
-  do a = 1, nV
-    do i = 1, nO
-      e = e + 2d0 * d_cc_space_f_vo%f(a,i) * t1%f(i,a)
-    enddo
-  enddo
-  !$omp end do nowait
-  !$omp do
-  do b = 1, nV
-    do a = 1, nV
-      do j = 1, nO
-        do i = 1, nO
-          e = e + tau_x%f(i,j,a,b) * d_cc_space_v_oovv%f(i,j,a,b)
-       enddo
-      enddo
-    enddo
-  enddo
-  !$omp end do nowait
-  !$omp critical
-  energy = energy + e
-  !$omp end critical
-  !$omp end parallel
-!
-!
-!   call gpu_ddot(blas_handle, nO*nO*nV*nV*1_8, tau_x, 1, d_cc_space_v_oovv, 1, energy)
-!   call gpu_ddot(blas_handle, nO*nV*1_8, d_cc_space_f_vo, 1, t1, 1, e)
-!   energy = energy + 2.d0*e
+!  energy = 0d0
+!  !$omp parallel &
+!  !$omp shared(nO,nV,energy,tau_x,t1,&
+!  !$omp d_cc_space_f_vo,d_cc_space_v_oovv) &
+!  !$omp private(i,j,a,b,e) &
+!  !$omp default(none)
+!  e = 0d0
+!  !$omp do
+!  do a = 1, nV
+!    do i = 1, nO
+!      e = e + 2d0 * d_cc_space_f_vo%f(a,i) * t1%f(i,a)
+!    enddo
+!  enddo
+!  !$omp end do nowait
+!  !$omp do
+!  do b = 1, nV
+!    do a = 1, nV
+!      do j = 1, nO
+!        do i = 1, nO
+!          e = e + tau_x%f(i,j,a,b) * d_cc_space_v_oovv%f(i,j,a,b)
+!       enddo
+!      enddo
+!    enddo
+!  enddo
+!  !$omp end do nowait
+!  !$omp critical
+!  energy = energy + e
+!  !$omp end critical
+!  !$omp end parallel
+
+
+   call gpu_ddot(blas_handle, nO*nO*nV*nV*1_8, tau_x, 1, d_cc_space_v_oovv, 1, energy)
+   call gpu_ddot(blas_handle, nO*nV*1_8, d_cc_space_f_vo, 1, t1, 1, e)
+   energy = energy + 2.d0*e
 
 end
 
diff --git a/src/gpu/gpu.irp.f b/src/gpu/gpu.irp.f
new file mode 100644
index 00000000..e91d66f5
--- /dev/null
+++ b/src/gpu/gpu.irp.f
@@ -0,0 +1,11 @@
+use gpu
+
+BEGIN_PROVIDER [ type(gpu_blas), blas_handle ]
+ implicit none
+ BEGIN_DOC
+ ! Handle for cuBLAS or RocBLAS
+ END_DOC
+ call gpu_blas_create(blas_handle)
+END_PROVIDER
+
+