This commit is contained in:
mtgu0705
2025-01-03 12:54:43 +08:00
parent e9520f14de
commit f52323fc1f
2 changed files with 2 additions and 2 deletions

View File

@@ -19,6 +19,8 @@ struct pk_i4_t
type data;
__host__ __device__ constexpr pk_i4_t() : data{type{}} {}
__host__ __device__ constexpr pk_i4_t(type init) : data{init} {}
__host__ __device__ constexpr operator float() const { return static_cast<float>(data); }
};
inline constexpr auto next_pow2(uint32_t x)

View File

@@ -105,8 +105,6 @@ int profile_gemm_b_scale(int argc, char* argv[])
using F32 = float;
using F16 = ck::half_t;
using BF16 = ck::bhalf_t;
using F8 = ck::f8_t;
using I4 = ck::pk_i4_t;
using Row = ck::tensor_layout::gemm::RowMajor;