Add missing macro #undef

b41fc37b · Po Yen, Chen · d03ecf40 · b41fc37b
Commit b41fc37b authored Jan 10, 2025 by Po Yen, Chen
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

example/ck_tile/18_paged_attention/include/kernel/attention_kernel.hpp ...le/18_paged_attention/include/kernel/attention_kernel.hpp +5 -0

No files found.
--- a/example/ck_tile/18_paged_attention/include/kernel/attention_kernel.hpp
+++ b/example/ck_tile/18_paged_attention/include/kernel/attention_kernel.hpp
@@ -1015,3 +1015,8 @@ __launch_bounds__(NUM_THREADS) void paged_attention_ll4mi_reduce_kernel(
    const float* __restrict__ fp8_out_scale_ptr){UNREACHABLE_CODE}

 #endif  // defined(__HIP__MI300_MI250__) TODO: Add NAVI support
+
+#undef WARP_SIZE
+#undef MAX
+#undef MIN
+#undef DIVIDE_ROUND_UP