Skip to content

Commit be1d8dc

Browse files
committed
Use smaller type for enums to reduce register usage
1 parent ff81c72 commit be1d8dc

File tree

2 files changed

+6
-8
lines changed

2 files changed

+6
-8
lines changed

include/cuco/detail/equal_wrapper.cuh

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -17,18 +17,17 @@
1717

1818
#include <cuco/detail/bitwise_compare.cuh>
1919

20-
#include <cstddef>
20+
#include <cuda/std/cstdint>
2121

22-
namespace cuco {
23-
namespace detail {
22+
namespace cuco::detail {
2423

2524
/**
2625
* @brief Enum of equality comparison results
2726
*/
2827
// ENUM VALUE MATTERS, DO NOT CHANGE
29-
enum class equal_result : int32_t { UNEQUAL = 0, EQUAL = 1, EMPTY = 2, AVAILABLE = 3 };
28+
enum class equal_result : cuda::std::int8_t { UNEQUAL = 0, EQUAL = 1, EMPTY = 2, AVAILABLE = 3 };
3029

31-
enum class is_insert : bool { YES, NO };
30+
enum class is_insert : cuda::std::int8_t { YES, NO };
3231

3332
/**
3433
* @brief Key equality wrapper.
@@ -116,5 +115,4 @@ struct equal_wrapper {
116115
}
117116
};
118117

119-
} // namespace detail
120-
} // namespace cuco
118+
} // namespace cuco::detail

include/cuco/detail/open_addressing/open_addressing_ref_impl.cuh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ namespace cuco {
4242
namespace detail {
4343

4444
/// Three-way insert result enum
45-
enum class insert_result : cuda::std::int32_t { CONTINUE = 0, SUCCESS = 1, DUPLICATE = 2 };
45+
enum class insert_result : cuda::std::int8_t { CONTINUE = 0, SUCCESS = 1, DUPLICATE = 2 };
4646

4747
/**
4848
* @brief Helper struct to store intermediate bucket probing results.

0 commit comments

Comments
 (0)