mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-11 17:00:18 +00:00
fixed amd build
This commit is contained in:
@@ -2,9 +2,9 @@
|
|||||||
#define CK_COMMON_HEADER_HPP
|
#define CK_COMMON_HEADER_HPP
|
||||||
|
|
||||||
#include "config.hpp"
|
#include "config.hpp"
|
||||||
|
#include "utility.hpp"
|
||||||
#include "integral_constant.hpp"
|
#include "integral_constant.hpp"
|
||||||
#include "math.hpp"
|
#include "math.hpp"
|
||||||
#include "utility.hpp"
|
|
||||||
#include "vector_type.hpp"
|
#include "vector_type.hpp"
|
||||||
#include "Sequence.hpp"
|
#include "Sequence.hpp"
|
||||||
#include "Array.hpp"
|
#include "Array.hpp"
|
||||||
@@ -12,4 +12,8 @@
|
|||||||
#include "functional2.hpp"
|
#include "functional2.hpp"
|
||||||
#include "functional3.hpp"
|
#include "functional3.hpp"
|
||||||
|
|
||||||
|
#if CK_USE_AMD_INLINE_ASM
|
||||||
|
#include "amd_inline_asm.hpp"
|
||||||
|
#endif
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
@@ -18,10 +18,6 @@ typedef float float4_t __attribute__((ext_vector_type(4)));
|
|||||||
|
|
||||||
using index_t = uint32_t;
|
using index_t = uint32_t;
|
||||||
|
|
||||||
__device__ index_t get_thread_local_1d_id() { return threadIdx.x; }
|
|
||||||
|
|
||||||
__device__ index_t get_block_1d_id() { return blockIdx.x; }
|
|
||||||
|
|
||||||
__device__ void fused_multiply_accumulate(float& d, const float& s0, const float& s1)
|
__device__ void fused_multiply_accumulate(float& d, const float& s0, const float& s1)
|
||||||
{
|
{
|
||||||
d += s0 * s1;
|
d += s0 * s1;
|
||||||
|
|||||||
@@ -8,12 +8,9 @@ struct integral_constant
|
|||||||
{
|
{
|
||||||
static constexpr T value = v;
|
static constexpr T value = v;
|
||||||
typedef T value_type;
|
typedef T value_type;
|
||||||
typedef integral_constant type; // using injected-class-name
|
typedef integral_constant type;
|
||||||
__host__ __device__ constexpr operator value_type() const noexcept { return value; }
|
__host__ __device__ constexpr operator value_type() const noexcept { return value; }
|
||||||
__host__ __device__ constexpr value_type operator()() const noexcept
|
__host__ __device__ constexpr value_type operator()() const noexcept { return value; }
|
||||||
{
|
|
||||||
return value;
|
|
||||||
} // since c++14
|
|
||||||
};
|
};
|
||||||
|
|
||||||
template <class T, T X, T Y>
|
template <class T, T X, T Y>
|
||||||
|
|||||||
@@ -2,6 +2,7 @@
|
|||||||
#define CK_MATH_HPP
|
#define CK_MATH_HPP
|
||||||
|
|
||||||
#include "config.hpp"
|
#include "config.hpp"
|
||||||
|
#include "integral_constant.hpp"
|
||||||
|
|
||||||
namespace ck {
|
namespace ck {
|
||||||
namespace math {
|
namespace math {
|
||||||
|
|||||||
Reference in New Issue
Block a user