diff --git a/library/src/amd_detail/hipfft.cpp b/library/src/amd_detail/hipfft.cpp index ef891a0..c2f7036 100644 --- a/library/src/amd_detail/hipfft.cpp +++ b/library/src/amd_detail/hipfft.cpp @@ -403,8 +403,9 @@ static void set_bricks(const std::vector& length, // that's bigger than 1 auto split_dim_iter = std::find_if( length_with_batch.rbegin(), length_with_batch.rend(), [](size_t len) { return len > 1; }); + // if all dimensions are 1 there's nothing to split if(split_dim_iter == length_with_batch.rend()) - throw HIPFFT_INVALID_VALUE; + return; size_t split_dim_idx = std::distance(length_with_batch.begin(), split_dim_iter.base()) - 1; for(size_t i = 0; i < bricks.size(); ++i)