Skip to content

GitLab

Commit c1d9c0ae authored by aska-0096's avatar aska-0096
Browse files

add some instances for AIT

parent 553cfa68
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
example/30_grouped_conv_fwd_multiple_d/run_grouped_conv_fwd_bias_relu_add_wmma_example.inc
View file @ c1d9c0ae
...@@ -35,6 +35,7 @@ using ResidualLayout = typename LayoutSettingSelector<NDimSpatial>::ResidualLayo ...@@ -35,6 +35,7 @@ using ResidualLayout = typename LayoutSettingSelector<NDimSpatial>::ResidualLayo
// clang-format off // clang-format off
template <ck::index_t NDimSpatial> template <ck::index_t NDimSpatial>
using DeviceConvFwdInstances = std::tuple< using DeviceConvFwdInstances = std::tuple<
#if 0
// Instances provide to AIT Fp16 // Instances provide to AIT Fp16
ck::tensor_operation::device::DeviceGroupedConvFwdMultipleD_Wmma_CShuffle< ck::tensor_operation::device::DeviceGroupedConvFwdMultipleD_Wmma_CShuffle<
NDimSpatial, NDimSpatial,
...@@ -157,7 +158,8 @@ using DeviceConvFwdInstances = std::tuple< ...@@ -157,7 +158,8 @@ using DeviceConvFwdInstances = std::tuple<
S<1, 64, 1>, S<1, 0, 2>, S<1, 0, 2>, 2, 8, 8, true, S<1, 64, 1>, S<1, 0, 2>, S<1, 0, 2>, 2, 8, 8, true,
S<1, 64, 1>, S<1, 0, 2>, S<1, 0, 2>, 2, 8, 8, true, S<1, 64, 1>, S<1, 0, 2>, S<1, 0, 2>, 2, 8, 8, true,
4, 1, S<1, 64, 1, 1>, 8> 4, 1, S<1, 64, 1, 1>, 8>
#if 0 #endif
#if 1
// GEMM_N = 16 // GEMM_N = 16
// K0 = 8 // K0 = 8
ck::tensor_operation::device::DeviceGroupedConvFwdMultipleD_Wmma_CShuffle< ck::tensor_operation::device::DeviceGroupedConvFwdMultipleD_Wmma_CShuffle<
...@@ -225,7 +227,7 @@ using DeviceConvFwdInstances = std::tuple< ...@@ -225,7 +227,7 @@ using DeviceConvFwdInstances = std::tuple<
2, 1, // MRepeat x NRepeat 2, 1, // MRepeat x NRepeat
S<1, 64, 1>, S<1, 0, 2>, S<1, 0, 2>, 2, 16, 16, true, // ABlockTransferThreadClusterLengths_AK0_M_AK1, ArrangeOrder, SrcAccessOrder, VectorDim, SrcScalarPerVector, DstScalarPerVector_AK1, LdsExtraM S<1, 64, 1>, S<1, 0, 2>, S<1, 0, 2>, 2, 16, 16, true, // ABlockTransferThreadClusterLengths_AK0_M_AK1, ArrangeOrder, SrcAccessOrder, VectorDim, SrcScalarPerVector, DstScalarPerVector_AK1, LdsExtraM
S<1, 16, 4>, S<1, 0, 2>, S<1, 0, 2>, 2, 4, 4, true, S<1, 16, 4>, S<1, 0, 2>, S<1, 0, 2>, 2, 4, 4, true,
2, 1, S<1, 64, 1, 1>, 16> 2, 1, S<1, 64, 1, 1>, 16>,
// K0 = 2 // K0 = 2
ck::tensor_operation::device::DeviceGroupedConvFwdMultipleD_Wmma_CShuffle< ck::tensor_operation::device::DeviceGroupedConvFwdMultipleD_Wmma_CShuffle<
NDimSpatial, NDimSpatial,
......
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