Skip to content

GitLab

"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "b4b8c723d377caae2f4243cf8fc711eab71736a1"
Commit cef44211 authored by letaoqin's avatar letaoqin
Browse files

rename the base class and file name

parent d8998cbb
Hide 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, ...@@ -31,7 +31,7 @@ template <index_t NumDimG,
typename B1ElementwiseOperation, typename B1ElementwiseOperation,
typename C1DEElementwiseOperation, typename C1DEElementwiseOperation,
MaskingSpecialization MaskingSpec> MaskingSpecialization MaskingSpec>
struct DeviceBatchedMultiheadAttentionForward : public BaseOperator struct DeviceBatchedMultiheadAttentionInfer : public BaseOperator
{ {
virtual std::unique_ptr<BaseArgument> MakeArgumentPointer( virtual std::unique_ptr<BaseArgument> MakeArgumentPointer(
const void* p_a, 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, ...@@ -30,7 +30,7 @@ template <index_t NumDimG,
typename B1ElementwiseOperation, typename B1ElementwiseOperation,
typename CElementwiseOperation, typename CElementwiseOperation,
MaskingSpecialization MaskingSpec> MaskingSpecialization MaskingSpec>
struct DeviceGroupedMultiheadAttentionForward : public BaseOperator struct DeviceGroupedMultiheadAttentionInfer : public BaseOperator
{ {
struct ProblemDesc 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, ...@@ -206,7 +206,7 @@ template <index_t NumDimG,
int D0sTransferSrcScalarPerVector = 4, int D0sTransferSrcScalarPerVector = 4,
LoopScheduler LoopSched = LoopScheduler::Default> LoopScheduler LoopSched = LoopScheduler::Default>
struct DeviceBatchedMultiheadAttentionForward_Xdl struct DeviceBatchedMultiheadAttentionForward_Xdl
: public DeviceBatchedMultiheadAttentionForward<NumDimG, : public DeviceBatchedMultiheadAttentionInfer<NumDimG,
NumDimM, NumDimM,
NumDimN, NumDimN,
NumDimK, 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, ...@@ -195,7 +195,7 @@ template <index_t NumDimG,
MaskingSpecialization MaskingSpec, MaskingSpecialization MaskingSpec,
LoopScheduler LoopSched = LoopScheduler::Default> LoopScheduler LoopSched = LoopScheduler::Default>
struct DeviceGroupedMultiheadAttentionForward_Xdl struct DeviceGroupedMultiheadAttentionForward_Xdl
: public DeviceGroupedMultiheadAttentionForward<NumDimG, : public DeviceGroupedMultiheadAttentionInfer<NumDimG,
NumDimM, NumDimM,
NumDimN, NumDimN,
NumDimK, 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