Skip to content

GitLab

Commit e6f4653a authored by Bartlomiej Kocot's avatar Bartlomiej Kocot
Browse files

Fix client example

parent dc045e2b
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 8 deletions
client_example/11_grouped_conv_bwd_weight/grouped_conv1d_bwd_weight_fp16.cpp
View file @ e6f4653a
......@@ -22,9 +22,9 @@ static constexpr ck::index_t C = 192;
static constexpr ck::index_t X = 3;
static constexpr ck::index_t Wi = 28;
static constexpr ck::index_t Wo = 28;
static constexpr std::array<ck::index_t, NDimSpatial + 3> input_strides{
static constexpr std::array<ck::index_t, NumDimSpatial + 3> input_strides{
G * N * Wi * C, N* Wi* C, Wi* C, C, 1};
static constexpr std::array<ck::index_t, NDimSpatial + 3> output_strides{
static constexpr std::array<ck::index_t, NumDimSpatial + 3> output_strides{
G * N * Wo * K, N* Wo* K, Wo* K, K, 1};
int main()
......
client_example/11_grouped_conv_bwd_weight/grouped_conv2d_bwd_weight_fp16.cpp
View file @ e6f4653a
......@@ -25,9 +25,9 @@ static constexpr ck::index_t Hi = 28;
static constexpr ck::index_t Wi = 28;
static constexpr ck::index_t Ho = 28;
static constexpr ck::index_t Wo = 28;
static constexpr std::array<ck::index_t, NDimSpatial + 3> input_strides{
static constexpr std::array<ck::index_t, NumDimSpatial + 3> input_strides{
G * N * Hi * Wi * C, N* Hi* Wi* C, Hi* Wi* C, Wi* C, C, 1};
static constexpr std::array<ck::index_t, NDimSpatial + 3> output_strides{
static constexpr std::array<ck::index_t, NumDimSpatial + 3> output_strides{
G * N * Ho * Wo * K, N* Ho* Wo* K, Ho* Wo* K, Wo* K, K, 1};
int main()
......
client_example/11_grouped_conv_bwd_weight/grouped_conv3d_bwd_weight_fp16.cpp
View file @ e6f4653a
......@@ -28,9 +28,9 @@ static constexpr ck::index_t Wi = 3;
static constexpr ck::index_t Do = 28;
static constexpr ck::index_t Ho = 28;
static constexpr ck::index_t Wo = 3;
static constexpr std::array<ck::index_t, NDimSpatial + 3> input_strides{
static constexpr std::array<ck::index_t, NumDimSpatial + 3> input_strides{
G * N * Di * Hi * Wi * C, N* Di* Hi* Wi* C, Di* Hi* Wi* C, Hi* Wi* C, Wi* C, C, 1};
static constexpr std::array<ck::index_t, NDimSpatial + 3> output_strides{
static constexpr std::array<ck::index_t, NumDimSpatial + 3> output_strides{
G * N * Do * Ho * Wo * K, N* Do* Ho* Wo* K, Do* Ho* Wo* K, Ho* Wo* K, Wo* K, K, 1};
int main()
......
client_example/11_grouped_conv_bwd_weight/grouped_conv3d_bwd_weight_fp32.cpp
View file @ e6f4653a
......@@ -28,9 +28,9 @@ static constexpr ck::index_t Wi = 3;
static constexpr ck::index_t Do = 28;
static constexpr ck::index_t Ho = 28;
static constexpr ck::index_t Wo = 3;
static constexpr std::array<ck::index_t, NDimSpatial + 3> input_strides{
static constexpr std::array<ck::index_t, NumDimSpatial + 3> input_strides{
G * N * Di * Hi * Wi * C, N* Di* Hi* Wi* C, Di* Hi* Wi* C, Hi* Wi* C, Wi* C, C, 1};
static constexpr std::array<ck::index_t, NDimSpatial + 3> output_strides{
static constexpr std::array<ck::index_t, NumDimSpatial + 3> output_strides{
G * N * Do * Ho * Wo * K, N* Do* Ho* Wo* K, Do* Ho* Wo* K, Ho* Wo* K, Wo* K, K, 1};
int main()
......
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