Skip to content

GitLab

Commit ed6c3d2d authored by ltqin's avatar ltqin
Browse files

regular code

parent 73156add
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
include/ck/tensor_operation/gpu/block/blockwise_softmax_v1.hpp
View file @ ed6c3d2d
...@@ -22,6 +22,10 @@ struct BlockwiseSoftmax_V1 ...@@ -22,6 +22,10 @@ struct BlockwiseSoftmax_V1
{ {
static_assert(MRepeat == 1, "Now MRepeat must equal 1"); static_assert(MRepeat == 1, "Now MRepeat must equal 1");
static constexpr auto I0 = Number<0>{};
static constexpr auto I1 = Number<1>{};
static constexpr auto I2 = Number<2>{};
static constexpr index_t MThreadSliceSize = 1;
static constexpr index_t WaveSize = 64; static constexpr index_t WaveSize = 64;
struct BlockToMKMap_M0_K_M1Adapt struct BlockToMKMap_M0_K_M1Adapt
...@@ -36,10 +40,6 @@ struct BlockwiseSoftmax_V1 ...@@ -36,10 +40,6 @@ struct BlockwiseSoftmax_V1
return make_tuple(m, k); return make_tuple(m, k);
} }
}; };
static constexpr auto I0 = Number<0>{};
static constexpr auto I1 = Number<1>{};
static constexpr auto I2 = Number<2>{};
static constexpr index_t MThreadSliceSize = 1;
constexpr static auto in_thread_desc = make_naive_tensor_descriptor_packed( constexpr static auto in_thread_desc = make_naive_tensor_descriptor_packed(
make_tuple(Number<MRepeat>{}, Number<NRepeat>{}, Number<RegSizePerXdlops>{})); make_tuple(Number<MRepeat>{}, Number<NRepeat>{}, Number<RegSizePerXdlops>{}));
......
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