diff options
| author |  Neïl Zaim <49716072+NeilZaim@users.noreply.github.com>
| 2021-10-27 17:13:23 +0200 |
|---|---|---|
| committer |  GitHub <noreply@github.com>
| 2021-10-27 08:13:23 -0700 |
| commit | cedd0b09a5c54902efa2731ec315ae25f6e2b2d6 (patch) | |
| tree | 8cb8806ae78fe12bcaf5957b316ccc1058c085f1 /Source/Particles/Collision/BinaryCollision/BinaryCollision.H | |
| parent | fa35a250d9cf3bb047ece81f5e138ce782fc18e0 (diff) | |
| download | WarpX-cedd0b09a5c54902efa2731ec315ae25f6e2b2d6.tar.gz WarpX-cedd0b09a5c54902efa2731ec315ae25f6e2b2d6.tar.zst WarpX-cedd0b09a5c54902efa2731ec315ae25f6e2b2d6.zip | |
Fix BinaryCollision on GPU (#2504)
Diffstat (limited to 'Source/Particles/Collision/BinaryCollision/BinaryCollision.H')
| -rw-r--r-- | Source/Particles/Collision/BinaryCollision/BinaryCollision.H | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/Source/Particles/Collision/BinaryCollision/BinaryCollision.H b/Source/Particles/Collision/BinaryCollision/BinaryCollision.H index bfb0d4c97..f85e67901 100644 --- a/Source/Particles/Collision/BinaryCollision/BinaryCollision.H +++ b/Source/Particles/Collision/BinaryCollision/BinaryCollision.H @@ -291,8 +291,9 @@ public: // Start indices of the pairs in a cell. Will be used for particle creation. amrex::Gpu::DeviceVector<index_type> pair_offsets(n_cells_products); - const auto n_total_pairs = amrex::Scan::ExclusiveSum(n_cells_products, - p_n_pairs_in_each_cell, pair_offsets.data()); + const index_type n_total_pairs = (n_cells_products == 0) ? 0: + amrex::Scan::ExclusiveSum(n_cells_products, + p_n_pairs_in_each_cell, pair_offsets.data()); index_type* AMREX_RESTRICT p_pair_offsets = pair_offsets.dataPtr(); // mask: equal to 1 if particle creation occurs for a given pair, 0 otherwise @@ -444,8 +445,9 @@ public: // Start indices of the pairs in a cell. Will be used for particle creation amrex::Gpu::DeviceVector<index_type> pair_offsets(n_cells_products); - auto n_total_pairs = amrex::Scan::ExclusiveSum(n_cells_products, - p_n_pairs_in_each_cell, pair_offsets.data()); + const index_type n_total_pairs = (n_cells_products == 0) ? 0: + amrex::Scan::ExclusiveSum(n_cells_products, + p_n_pairs_in_each_cell, pair_offsets.data()); index_type* AMREX_RESTRICT p_pair_offsets = pair_offsets.dataPtr(); // mask: equal to 1 if particle creation occurs for a given pair, 0 otherwise |