Skip to content

Commit

Permalink
Navi32 multiVF failing test (#455) (#459)
Browse files Browse the repository at this point in the history
* Fix test device_radix_sort

* Remove comment
  • Loading branch information
RobsonRLemos authored Jul 31, 2023
1 parent e0311e4 commit b54aaa7
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions test/rocprim/test_device_radix_sort.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -716,6 +716,7 @@ inline void sort_keys_over_4g()

key_type* d_keys_input_output{};
size_t key_type_storage_bytes = size * sizeof(key_type);

HIP_CHECK(test_common_utils::hipMallocHelper(&d_keys_input_output, key_type_storage_bytes));
HIP_CHECK(hipMemcpy(d_keys_input_output,
keys_input.data(),
Expand All @@ -738,12 +739,12 @@ inline void sort_keys_over_4g()
hipDeviceProp_t prop;
HIP_CHECK(hipGetDeviceProperties(&prop, device_id));

size_t total_storage_bytes = key_type_storage_bytes + temporary_storage_bytes;
if (total_storage_bytes > prop.totalGlobalMem) {
size_t total_storage_bytes = key_type_storage_bytes + temporary_storage_bytes;
if (total_storage_bytes > (static_cast<size_t>(prop.totalGlobalMem * 0.90))) {
HIP_CHECK(hipFree(d_keys_input_output));
GTEST_SKIP() << "Test case device memory requirement (" << total_storage_bytes << " bytes) exceeds available memory on current device ("
<< prop.totalGlobalMem << " bytes). Skipping test";
}
}

void* d_temporary_storage;
HIP_CHECK(test_common_utils::hipMallocHelper(&d_temporary_storage, temporary_storage_bytes));
Expand Down

0 comments on commit b54aaa7

Please sign in to comment.