enable bf16 atomic add on gfx950

ca15fa77 · illsilin · 140d2fa6 · ca15fa77
Commit ca15fa77 authored Oct 14, 2024 by illsilin
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

include/ck/utility/amd_buffer_addressing.hpp include/ck/utility/amd_buffer_addressing.hpp +1 -1

No files found.
--- a/include/ck/utility/amd_buffer_addressing.hpp
+++ b/include/ck/utility/amd_buffer_addressing.hpp
@@ -578,7 +578,7 @@ __device__ void amd_global_atomic_add_impl(const typename vector_type<T, N>::typ
                                                      tmp.template AsType<half2_t>()[i]);
        });
    }
-#if defined(__gfx942__)
+#if defined(__gfx942__) || defined(__gfx950__)
    else if constexpr(is_same<T, bhalf_t>::value)
    {
        vector_type<bhalf_t, N> tmp{src_thread_data};