(iterations: int, workspace_index: int = 0)
| 901 | _cuda_success(err, "Error on freeing memory") |
| 902 | |
| 903 | def _loop_and_call_kernel(iterations: int, workspace_index: int = 0) -> int: |
| 904 | for _ in range(iterations): |
| 905 | current_workspace = workspaces[workspace_index] |
| 906 | callable(*current_workspace.args, **current_workspace.kwargs) |
| 907 | workspace_index = (workspace_index + 1) % workspace_count |
| 908 | return workspace_index |
| 909 | |
| 910 | # Create CUDA events for timing |
| 911 | err, start_event = cuda_driver.cuEventCreate( |
no test coverage detected