Skip to content

GitLab

"llm/git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "b24e8d17b29246cbf520cffc48eac374ec245e48"
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
{
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;
struct BlockToMKMap_M0_K_M1Adapt
......@@ -36,10 +40,6 @@ struct BlockwiseSoftmax_V1
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(
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