|
7 | 7 |
|
8 | 8 |
|
9 | 9 | class _KirchhoffCudaHelper: |
10 | | - """A helper class to perform Kirchhoff demigration/migration using Numba CUDA. |
| 10 | + """Helper class for Kirchhoff operator using Numba CUDA. |
11 | 11 |
|
12 | | - This class provides methods to compute the forward and adjoint operations for the |
13 | | - Kirchhoff operator, utilizing GPU acceleration through Numba's CUDA capabilities. |
| 12 | + This class provides methods to compute the forward and adjoint operations of the |
| 13 | + Kirchhoff operator utilizing GPU acceleration through Numba's CUDA capabilities. |
14 | 14 |
|
15 | 15 | Parameters |
16 | 16 | ---------- |
@@ -39,7 +39,8 @@ def _grid_setup(self): |
39 | 39 | This method configures the number of blocks and threads per block for |
40 | 40 | CUDA kernels, depending on the number of sources and receivers. |
41 | 41 | """ |
42 | | - # use warp size as number of threads per block |
| 42 | + # use half of warp size as number of threads per block |
| 43 | + # although this will be ideally lifted up to the user |
43 | 44 | current_device = cuda.get_current_device() |
44 | 45 | warp = current_device.WARP_SIZE // 2 |
45 | 46 | self.num_threads_per_blocks = (warp, warp) |
@@ -339,7 +340,7 @@ def _call_kinematic(self, opt, *inputs): |
339 | 340 | return y_d |
340 | 341 |
|
341 | 342 | def _call_dynamic(self, opt, *inputs): |
342 | | - """Synamic computations using CUDA. |
| 343 | + """Dynamic computations using CUDA. |
343 | 344 |
|
344 | 345 | This method handles data preparation and execution of CUDA kernels |
345 | 346 | for both forward and adjoint operations of dynamic operator. |
|
0 commit comments