AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-511125 - Don't use device# as it varies BlockDim & GridDim.

Browse Source 
Change-Id: I3ddd0fc616c28adc2d35e64550bac4ca1c09bd6a
This commit is contained in:
Jaydeep Patel
2025-01-23 12:33:55 +00:00
committed by Rakesh Roy
parent 8ea926725b
commit 28e93a178f
1 changed files with 2 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
catch/unit/cooperativeGrps/multi_grid_group.cc
+2 -7
View File
@@ -135,13 +135,8 @@ static void get_multi_grid_dims(dim3& grid_dim, dim3& block_dim, unsigned int de
dim3(1, sm, 2),
dim3(3, 3, 3)};
if (test_case < 10) {
grid_dim = grid_dim_values[test_case % grid_dim_values.size()];
block_dim = block_dim_values[test_case % block_dim_values.size()];
} else {
grid_dim = grid_dim_values[(test_case + device) % grid_dim_values.size()];
block_dim = block_dim_values[(test_case + device) % block_dim_values.size()];
}
grid_dim = grid_dim_values[test_case % grid_dim_values.size()];
block_dim = block_dim_values[test_case % block_dim_values.size()];
}
/**