Skip to content

GitLab

Commit cef44211 authored by letaoqin's avatar letaoqin
Browse files

rename the base class and file name

parent d8998cbb
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
include/ck/tensor_operation/gpu/device/device_batched_multiple_head_flash_attention_fwd.hpp → include/ck/tensor_operation/gpu/device/device_batched_mha_infer.hpp
View file @ cef44211
......@@ -31,7 +31,7 @@ template <index_t NumDimG,
typename B1ElementwiseOperation,
typename C1DEElementwiseOperation,
MaskingSpecialization MaskingSpec>
struct DeviceBatchedMultiheadAttentionForward : public BaseOperator
struct DeviceBatchedMultiheadAttentionInfer : public BaseOperator
{
virtual std::unique_ptr<BaseArgument> MakeArgumentPointer(
const void* p_a,
......
include/ck/tensor_operation/gpu/device/device_grouped_multiple_head_flash_attention_fwd.hpp → include/ck/tensor_operation/gpu/device/device_grouped_mha_infer.hpp
View file @ cef44211
......@@ -30,7 +30,7 @@ template <index_t NumDimG,
typename B1ElementwiseOperation,
typename CElementwiseOperation,
MaskingSpecialization MaskingSpec>
struct DeviceGroupedMultiheadAttentionForward : public BaseOperator
struct DeviceGroupedMultiheadAttentionInfer : public BaseOperator
{
struct ProblemDesc
{
......
include/ck/tensor_operation/gpu/device/impl/device_batched_mha_fwd_xdl_cshuffle.hpp
View file @ cef44211
......@@ -206,7 +206,7 @@ template <index_t NumDimG,
int D0sTransferSrcScalarPerVector = 4,
LoopScheduler LoopSched = LoopScheduler::Default>
struct DeviceBatchedMultiheadAttentionForward_Xdl
: public DeviceBatchedMultiheadAttentionForward<NumDimG,
: public DeviceBatchedMultiheadAttentionInfer<NumDimG,
NumDimM,
NumDimN,
NumDimK,
......
include/ck/tensor_operation/gpu/device/impl/device_grouped_mha_fwd_xdl_cshuffle.hpp
View file @ cef44211
......@@ -195,7 +195,7 @@ template <index_t NumDimG,
MaskingSpecialization MaskingSpec,
LoopScheduler LoopSched = LoopScheduler::Default>
struct DeviceGroupedMultiheadAttentionForward_Xdl
: public DeviceGroupedMultiheadAttentionForward<NumDimG,
: public DeviceGroupedMultiheadAttentionInfer<NumDimG,
NumDimM,
NumDimN,
NumDimK,
......
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