mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-11 17:00:18 +00:00
32 lines
459 B
C++
32 lines
459 B
C++
#ifndef CK_AMD_ADDRESS_SPACE_HPP
|
|
#define CK_AMD_ADDRESS_SPACE_HPP
|
|
|
|
#include "config.hpp"
|
|
|
|
namespace ck {
|
|
|
|
enum AddressSpaceEnum_t
|
|
{
|
|
Generic,
|
|
Global,
|
|
Lds,
|
|
Sgpr,
|
|
Vgpr,
|
|
};
|
|
|
|
template <typename T>
|
|
__device__ T* cast_pointer_to_generic_address_space(T CONSTANT* p)
|
|
{
|
|
return (T*)p;
|
|
}
|
|
|
|
template <typename T>
|
|
__host__ __device__ T CONSTANT* cast_pointer_to_constant_address_space(T* p)
|
|
{
|
|
return (T CONSTANT*)p;
|
|
}
|
|
|
|
} // namespace ck
|
|
|
|
#endif
|