Working on fft on the device

2022-08-15 15:51:43 -05:00
parent f1112ab6b6
commit 46b8b00a4f
4 changed files with 34 additions and 2 deletions
--- a/lib/gpu/lal_amoeba_ext.cpp
+++ b/lib/gpu/lal_amoeba_ext.cpp
@ -162,6 +162,10 @@ void amoeba_gpu_compute_polar_real(int *host_amtype, int *host_amgroup, double *
                              eflag_in, vflag_in, eatom, vatom, aewald, felec, off2, tep_ptr);
 }

+void amoeba_compute_fft1d(void** in, void** out, const int mode) {
+  AMOEBAMF.compute_fft1d(in, out, mode);
+}
+
 double amoeba_gpu_bytes() {
  return AMOEBAMF.host_memory_usage();
 }