ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-24 23:05:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CK] s_prefetch unit test fixes.

Browse Source 
Signed-off-by: Michal Kulikowski <Michal.Kulikowski@amd.com>


[ROCm/composable_kernel commit: cd8af997e6]
This commit is contained in:
Michal Kulikowski
2025-11-10 11:19:37 +01:00
committed by Michał Kulikowski
parent 6c23879329
commit dd53cdad01
5 changed files with 184 additions and 248 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
include/ck/utility/amd_inline_asm.hpp
View File

@@ -432,28 +432,5 @@ __device__ void amd_assembly_outer_product_1x4(int8x16_t a,
}
#endif
// s_buffer_loads
inline __device__ int32_t
amd_assembly_s_buffer_load_b32(__amdgpu_buffer_rsrc_t src_wave_buffer_resource, unsigned int offset)
{
int32_t result;
asm volatile("s_buffer_load_b32 %0, %1, %2"
: "=s"(result)
: "s"(src_wave_buffer_resource), "s"(offset)
: "memory");
return result;
}
inline __device__ int32x2_t
amd_assembly_s_buffer_load_b64(__amdgpu_buffer_rsrc_t src_wave_buffer_resource, unsigned int offset)
{
int32x2_t result;
asm volatile("s_buffer_load_b64 %0, %1, %2"
: "=s"(result)
: "s"(src_wave_buffer_resource), "s"(offset)
: "memory");
return result;
}
} // namespace ck
#endif