"src/include/threadwise_gemm.hpp" did not exist on "abf75ac039420f7a4ab64a419416dd493b906742"
Commit dc60c09f authored by Paul's avatar Paul
Browse files

Fix warnings

parent d992494e
...@@ -48,6 +48,14 @@ inline auto gs_launch(std::size_t n, std::size_t local = 1024) ...@@ -48,6 +48,14 @@ inline auto gs_launch(std::size_t n, std::size_t local = 1024)
}; };
} }
// Workaround hcc's broken tile_static macro
#ifdef tile_static
#undef tile_static
#define MIGRAPH_DEVICE_SHARED __attribute__((tile_static))
#else
#define MIGRAPH_DEVICE_SHARED __shared__
#endif
} // namespace device } // namespace device
} // namespace gpu } // namespace gpu
} // namespace migraph } // namespace migraph
......
...@@ -83,7 +83,7 @@ inline auto binary_broadcast_vec(argument result, argument arg1, argument arg2) ...@@ -83,7 +83,7 @@ inline auto binary_broadcast_vec(argument result, argument arg1, argument arg2)
const std::size_t bdim_vec_len = bdim_len / vec_size; const std::size_t bdim_vec_len = bdim_len / vec_size;
launch(nglobal, nlocal)([=](auto idx) __device__ { launch(nglobal, nlocal)([=](auto idx) __device__ {
__shared__ vec4<type> buffer[2048 / vec_size]; MIGRAPH_DEVICE_SHARED vec4<type> buffer[2048 / vec_size];
// Load bias into LDS // Load bias into LDS
for(size_t i = idx.local; i < bdim_vec_len; i += nlocal) for(size_t i = idx.local; i < bdim_vec_len; i += nlocal)
{ {
...@@ -133,7 +133,7 @@ inline auto binary_broadcast(argument result, argument arg1, argument arg2) ...@@ -133,7 +133,7 @@ inline auto binary_broadcast(argument result, argument arg1, argument arg2)
const std::size_t n = output.size(); const std::size_t n = output.size();
launch(nglobal, nlocal)([=](auto idx) __device__ { launch(nglobal, nlocal)([=](auto idx) __device__ {
__shared__ type buffer[2048]; MIGRAPH_DEVICE_SHARED type buffer[2048];
// Load bias into LDS // Load bias into LDS
for(size_t i = idx.local; i < bdim_len; i += nlocal) for(size_t i = idx.local; i < bdim_len; i += nlocal)
{ {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment