Merge pull request #16 from BKP/develop

Fix signal calculation bug for fine-tiled unit tests

[ROCm/rocshmem commit: 7fd66663c5]
Este commit está contenido en:
Brandon Potter
2024-12-20 08:40:56 -06:00
cometido por GitHub
Se han modificado 2 ficheros con 2 adiciones y 1 borrados
@@ -141,7 +141,7 @@ int warp_signals_calculation(int grid_dim_x, int block_dim_x, size_t size) {
int warps_per_block {warpsPerBlock(block_dim_x)};
int total_num_warps_in_grid {grid_dim_x * warps_per_block};
int bytes_per_warp {WARP_SIZE * THREAD_TRANSFER_GRANULARITY};
int bytes_per_warp {min(block_dim_x, WARP_SIZE) * THREAD_TRANSFER_GRANULARITY};
int num_signals_for_one_full_iteration = total_num_warps_in_grid;
int num_signals_for_partial_last_iteration = ((partial_grid_last_iteration_data_size - THREAD_TRANSFER_GRANULARITY) + bytes_per_warp) / bytes_per_warp;
@@ -257,6 +257,7 @@ class IPCImplTiledFine : public ::testing::TestWithParam<std::tuple<int, int, in
int *dest = reinterpret_cast<int*>(ipc_impl_.ipc_bases[1]);
FN_T2 val_fn = kernel_put_with_signal_tiled_validator;
launch(val_fn, grid, block, dest, bytes);
ASSERT_EQ(*(dest + SIGNAL_OFFSET), 0);
}
mpi_.barrier();
return;