make sure to set F/B scales in the descriptor to match what is happening in Python

Author: oleksandr-pavlyk

native/fft2_cdp-in-c.c

@@ -86,6 +86,12 @@ int main() {
         status = DftiCommitDescriptor(hand);
         assert(status == 0);
 
+        status = DftiSetValue(hand, DFTI_FORWARD_SCALE, 1.0);
+        assert(status == 0);
+
+        status = DftiSetValue(hand, DFTI_BACKWARD_SCALE, 1.0/(N1 * N2));
+        assert(status == 0);
+
 //        status = DftiSetValue(hand, DFTI_COMPLEX_STORAGE, DFTI_COMPLEX_COMPLEX);
 //        assert(status == 0);
 

native/fft2_cdp-in-nc.c

@@ -68,9 +68,9 @@ int main() {
 
     warm_up_threads();
 
-    for(si=0; si < samps; time_tot=0, si++) {
+    for(si=0; si < samps; si++) {
 
-        for(it = -1; it <reps;  it++) {
+        for(time_tot=0, it = -1; it <reps;  it++) {
 
            cblas_zcopy(N, (void *) x, 1, (void *) buf, 1); /* buf = x */
            /* memcpy(buf, x, N*sizeof(MKL_Complex16)); */
@@ -88,6 +88,12 @@ int main() {
             status = DftiSetValue(hand, DFTI_INPUT_STRIDES, strides);
             assert(status == 0);
 
+            status = DftiSetValue(hand, DFTI_FORWARD_SCALE, 1.0);
+            assert(status == 0);
+
+            status = DftiSetValue(hand, DFTI_BACKWARD_SCALE, 1.0/(N1 * N2));
+            assert(status == 0);
+
             status = DftiCommitDescriptor(hand);
             assert(status == 0);
 

native/fft2_cdp-out-c.c

@@ -87,6 +87,12 @@ int main() {
         status = DftiSetValue(hand, DFTI_OUTPUT_STRIDES, strides);
         assert(status == 0);
 
+        status = DftiSetValue(hand, DFTI_FORWARD_SCALE, 1.0);
+        assert(status == 0);
+
+        status = DftiSetValue(hand, DFTI_BACKWARD_SCALE, 1.0/(N1 * N2));
+        assert(status == 0);
+
         status = DftiCommitDescriptor(hand);
         assert(status == 0);
 

native/fft2_cdp-out-nc.c

@@ -89,6 +89,12 @@ int main() {
             status = DftiSetValue(hand, DFTI_OUTPUT_STRIDES, strides);
             assert(status == 0);
 
+            status = DftiSetValue(hand, DFTI_FORWARD_SCALE, 1.0);
+            assert(status == 0);
+
+            status = DftiSetValue(hand, DFTI_BACKWARD_SCALE, 1.0/(N1 * N2));
+            assert(status == 0);
+
             status = DftiCommitDescriptor(hand);
             assert(status == 0);
 

native/fft3_cdp-in-c.c

@@ -79,6 +79,12 @@ int main() {
         status = DftiSetValue(hand, DFTI_INPUT_STRIDES, strides);
         assert(status == 0);
 
+        status = DftiSetValue(hand, DFTI_FORWARD_SCALE, 1.0);
+        assert(status == 0);
+
+        status = DftiSetValue(hand, DFTI_BACKWARD_SCALE, 1.0/(N1 * N2 * N3));
+        assert(status == 0);
+
         status = DftiCommitDescriptor(hand);
         assert(status == 0);
 

native/fft3_cdp-in-nc.c

@@ -89,6 +89,13 @@ int main() {
             status = DftiSetValue(hand, DFTI_INPUT_STRIDES, strides);
             assert(status == 0);
 
+            status = DftiSetValue(hand, DFTI_FORWARD_SCALE, 1.0);
+            assert(status == 0);
+
+            status = DftiSetValue(hand, DFTI_BACKWARD_SCALE, 1.0/(N1 * N2 * N3));
+            assert(status == 0);
+
+
             status = DftiCommitDescriptor(hand);
             assert(status == 0);
 

native/fft3_cdp-out-c.c

@@ -83,6 +83,12 @@ int main() {
         status = DftiSetValue(hand, DFTI_OUTPUT_STRIDES, strides);
         assert(status == 0);
 
+        status = DftiSetValue(hand, DFTI_FORWARD_SCALE, 1.0);
+        assert(status == 0);
+
+        status = DftiSetValue(hand, DFTI_BACKWARD_SCALE, 1.0/(N1 * N2 * N3));
+        assert(status == 0);
+
         status = DftiCommitDescriptor(hand);
         assert(status == 0);
 

native/fft3_cdp-out-nc.c

@@ -90,6 +90,12 @@ int main() {
             status = DftiSetValue(hand, DFTI_OUTPUT_STRIDES, strides);
             assert(status == 0);
 
+            status = DftiSetValue(hand, DFTI_FORWARD_SCALE, 1.0);
+            assert(status == 0);
+
+            status = DftiSetValue(hand, DFTI_BACKWARD_SCALE, 1.0/(N1 * N2 * N3));
+            assert(status == 0);
+
             status = DftiCommitDescriptor(hand);
             assert(status == 0);