In this example a three-dimensional complex-to-complex transform is applied to the input data. Afterwards an inverse transform is performed on the computed frequency domain representation.
All GPUs supported by CUDA Toolkit (https://developer.nvidia.com/cuda-gpus)
Linux
Windows
x86_64
ppc64le
arm64-sbsa
- A Linux/Windows system with recent NVIDIA drivers.
- CMake version 3.18 minimum
$ mkdir build
$ cd build
$ cmake ..
$ make
Make sure that CMake finds expected CUDA Toolkit. If that is not the case you can add argument -DCMAKE_CUDA_COMPILER=/path/to/cuda/bin/nvcc to cmake command.
$ ./bin/3d_c2c_example
Sample example output (batch_size=1):
Input array:
0.000000 + 0.000000j
1.000000 + -1.000000j
2.000000 + -2.000000j
3.000000 + -3.000000j
4.000000 + -4.000000j
5.000000 + -5.000000j
6.000000 + -6.000000j
7.000000 + -7.000000j
=====
Output array:
0.000000 + 0.000000j
8.000000 + -8.000000j
16.000000 + -16.000000j
24.000000 + -24.000000j
32.000000 + -32.000000j
40.000000 + -40.000000j
48.000000 + -48.000000j
56.000000 + -56.000000j
=====
Sample example output (batch_size=2):
Input array:
0.000000 + 0.000000j
1.000000 + -1.000000j
2.000000 + -2.000000j
3.000000 + -3.000000j
4.000000 + -4.000000j
5.000000 + -5.000000j
6.000000 + -6.000000j
7.000000 + -7.000000j
8.000000 + -8.000000j
9.000000 + -9.000000j
10.000000 + -10.000000j
11.000000 + -11.000000j
12.000000 + -12.000000j
13.000000 + -13.000000j
14.000000 + -14.000000j
15.000000 + -15.000000j
=====
Output array:
0.000000 + 0.000000j
8.000000 + -8.000000j
16.000000 + -16.000000j
24.000000 + -24.000000j
32.000000 + -32.000000j
40.000000 + -40.000000j
48.000000 + -48.000000j
56.000000 + -56.000000j
64.000000 + -64.000000j
72.000000 + -72.000000j
80.000000 + -80.000000j
88.000000 + -88.000000j
96.000000 + -96.000000j
104.000000 + -104.000000j
112.000000 + -112.000000j
120.000000 + -120.000000j
=====