Skip to content

Commit d224329

Browse files
committed
Build correct workDiv for HIP device.
1 parent 81cd2b7 commit d224329

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

device/alpaka/src/utils/utils.hpp

+2-1
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,8 @@ static constexpr std::size_t warpSize =
4040
template <typename TAcc>
4141
inline WorkDiv makeWorkDiv(Idx blocksPerGrid,
4242
Idx threadsPerBlockOrElementsPerThread) {
43-
if constexpr (::alpaka::accMatchesTags<TAcc, ::alpaka::TagGpuCudaRt>) {
43+
if constexpr (::alpaka::accMatchesTags<TAcc, ::alpaka::TagGpuCudaRt> ||
44+
::alpaka::accMatchesTags<TAcc, ::alpaka::TagGpuHipRt>) {
4445
const auto elementsPerThread = Idx{1};
4546
return WorkDiv{blocksPerGrid, threadsPerBlockOrElementsPerThread,
4647
elementsPerThread};

0 commit comments

Comments
 (0)