Commit 8e862b7b authored by aska-0096's avatar aska-0096
Browse files

Change GridwiseOp nam

parent c5fd087e
...@@ -526,7 +526,7 @@ struct DeviceBatchedContractionMultipleD_Wmma_CShuffle ...@@ -526,7 +526,7 @@ struct DeviceBatchedContractionMultipleD_Wmma_CShuffle
using BGridDesc = decltype(DeviceOp::MakeBGridDescriptor({}, {})); using BGridDesc = decltype(DeviceOp::MakeBGridDescriptor({}, {}));
// GridwiseOp // GridwiseOp
using GridwiseOp = GridwiseGemmMultipleD_k0mk1_k0nk1_mn_wmma_cshuffle< using GridwiseOp = GridwiseGemmMultipleD_Wmma<
// DataType Family // DataType Family
ADataType, ADataType,
BDataType, BDataType,
......
...@@ -286,7 +286,7 @@ struct DeviceBatchedGemmSoftmaxGemmPermute_Wmma_CShuffle ...@@ -286,7 +286,7 @@ struct DeviceBatchedGemmSoftmaxGemmPermute_Wmma_CShuffle
}; };
// GridwiseOp // GridwiseOp
using GridwiseOp = GridwiseBatchedGemmSoftmaxGemm_Wmma_CShuffle< using GridwiseOp = GridwiseBatchedGemmSoftmaxGemm_Wmma<
// DataType Family // DataType Family
ADataType, ADataType,
B0DataType, B0DataType,
......
...@@ -263,7 +263,7 @@ struct DeviceGemmMultipleD_Wmma_CShuffle : public DeviceGemmMultipleD<ALayout, ...@@ -263,7 +263,7 @@ struct DeviceGemmMultipleD_Wmma_CShuffle : public DeviceGemmMultipleD<ALayout,
using EGridDesc_M_N = decltype(MakeEGridDescriptor_M_N<ELayout>(1, 1, 1)); using EGridDesc_M_N = decltype(MakeEGridDescriptor_M_N<ELayout>(1, 1, 1));
// GridwiseOp // GridwiseOp
using GridwiseOp = GridwiseGemmMultipleD_k0mk1_k0nk1_mn_wmma_cshuffle< using GridwiseOp = GridwiseGemmMultipleD_Wmma<
// DataType Family // DataType Family
ADataType, ADataType,
BDataType, BDataType,
......
...@@ -222,7 +222,7 @@ struct DeviceGemmWmma_CShuffle : public DeviceGemm<ALayout, ...@@ -222,7 +222,7 @@ struct DeviceGemmWmma_CShuffle : public DeviceGemm<ALayout,
using CGridDesc_M_N = decltype(MakeCGridDescriptor_M_N(1, 1, 1)); using CGridDesc_M_N = decltype(MakeCGridDescriptor_M_N(1, 1, 1));
// GridwiseGemm // GridwiseGemm
using GridwiseGemm = GridwiseGemm_k0mk1_k0nk1_mn_wmma< using GridwiseGemm = GridwiseGemm_Wmma<
BlockSize, BlockSize,
ADataType, ADataType,
BDataType, BDataType,
......
...@@ -323,7 +323,7 @@ struct DeviceGroupedConvFwdMultipleD_Wmma_CShuffle ...@@ -323,7 +323,7 @@ struct DeviceGroupedConvFwdMultipleD_Wmma_CShuffle
using BGridDesc = decltype(DeviceOp::MakeBGridDescriptor<BLayout>({}, {})); using BGridDesc = decltype(DeviceOp::MakeBGridDescriptor<BLayout>({}, {}));
// GridwiseOp // GridwiseOp
using GridwiseOp = GridwiseGemmMultipleD_k0mk1_k0nk1_mn_wmma_cshuffle< using GridwiseOp = GridwiseGemmMultipleD_Wmma<
// DataType Family // DataType Family
ADataType, ADataType,
BDataType, BDataType,
......
...@@ -182,7 +182,7 @@ template <typename ADataType, ...@@ -182,7 +182,7 @@ template <typename ADataType,
index_t NumGemmKPrefetchStage = 1, index_t NumGemmKPrefetchStage = 1,
LoopScheduler LoopSched = make_default_loop_scheduler(), LoopScheduler LoopSched = make_default_loop_scheduler(),
PipelineVersion PipelineVer = PipelineVersion::v1> PipelineVersion PipelineVer = PipelineVersion::v1>
struct GridwiseBatchedGemmSoftmaxGemm_Wmma_CShuffle struct GridwiseBatchedGemmSoftmaxGemm_Wmma
{ {
static constexpr auto I0 = Number<0>{}; static constexpr auto I0 = Number<0>{};
static constexpr auto I1 = Number<1>{}; static constexpr auto I1 = Number<1>{};
......
...@@ -321,7 +321,7 @@ template < // DataType Family ...@@ -321,7 +321,7 @@ template < // DataType Family
index_t NumGemmKPrefetchStage = 1, index_t NumGemmKPrefetchStage = 1,
LoopScheduler LoopSched = make_default_loop_scheduler(), LoopScheduler LoopSched = make_default_loop_scheduler(),
PipelineVersion PipelineVer = PipelineVersion::v1> PipelineVersion PipelineVer = PipelineVersion::v1>
struct GridwiseGemmMultipleD_k0mk1_k0nk1_mn_wmma_cshuffle struct GridwiseGemmMultipleD_Wmma
{ {
static constexpr index_t NumDTensor = DsDataType::Size(); static constexpr index_t NumDTensor = DsDataType::Size();
......
...@@ -119,7 +119,7 @@ template <index_t BlockSize, ...@@ -119,7 +119,7 @@ template <index_t BlockSize,
index_t NumGemmKPrefetchStage = 1, index_t NumGemmKPrefetchStage = 1,
LoopScheduler LoopSched = make_default_loop_scheduler(), LoopScheduler LoopSched = make_default_loop_scheduler(),
PipelineVersion PipelineVer = PipelineVersion::v1> PipelineVersion PipelineVer = PipelineVersion::v1>
struct GridwiseGemm_k0mk1_k0nk1_mn_wmma struct GridwiseGemm_Wmma
{ {
static constexpr auto I0 = Number<0>{}; static constexpr auto I0 = Number<0>{};
static constexpr auto I1 = Number<1>{}; static constexpr auto I1 = Number<1>{};
......
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