Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "12dfba3d03f402c051e2129fa21f33264f4d26e5"
Commit 58584c29 authored by Chao Liu's avatar Chao Liu
Browse files

tweaking bwd data

parent e2b4c5b4
Show whitespace changes
Inline Side-by-side
Showing with 80 additions and 18 deletions
driver/src/conv_bwd_data_driver.cpp
View file @ 58584c29
...@@ -21,21 +21,52 @@ int main(int argc, char* argv[]) ...@@ -21,21 +21,52 @@ int main(int argc, char* argv[])
{ {
using namespace ck; using namespace ck;
#if 1 #if 0
constexpr index_t N = 8; // 1x1
constexpr index_t C = 128; constexpr index_t N = 256;
constexpr index_t HI = 16; constexpr index_t C = 1024;
constexpr index_t WI = 16; constexpr index_t HI = 8;
constexpr index_t K = 8; constexpr index_t WI = 8;
constexpr index_t Y = 2; constexpr index_t K = 1024;
constexpr index_t X = 2; constexpr index_t Y = 1;
constexpr index_t X = 1;
using ConvStrides = Sequence<4, 4>; using ConvStrides = Sequence<1, 1>;
using ConvDilations = Sequence<2, 2>; using ConvDilations = Sequence<1, 1>;
using LeftPads = Sequence<0, 0>; using LeftPads = Sequence<0, 0>;
using RightPads = Sequence<0, 0>; using RightPads = Sequence<0, 0>;
#elif 0 #elif 0
// 1x7
constexpr index_t N = 128;
constexpr index_t C = 1024;
constexpr index_t HI = 17;
constexpr index_t WI = 17;
constexpr index_t K = 1024;
constexpr index_t Y = 1;
constexpr index_t X = 7;
using ConvStrides = Sequence<1, 1>;
using ConvDilations = Sequence<1, 1>;
using LeftPads = Sequence<0, 3>;
using RightPads = Sequence<0, 3>;
#elif 1
// 3x3, 34x34
constexpr index_t N = 64;
constexpr index_t C = 256;
constexpr index_t HI = 34;
constexpr index_t WI = 34;
constexpr index_t K = 256;
constexpr index_t Y = 3;
constexpr index_t X = 3;
using ConvStrides = Sequence<1, 1>;
using ConvDilations = Sequence<1, 1>;
using LeftPads = Sequence<0, 0>;
using RightPads = Sequence<0, 0>;
#elif 1
// 3x3, 34x34 // 3x3, 34x34
constexpr index_t N = 64; constexpr index_t N = 64;
constexpr index_t C = 256; constexpr index_t C = 256;
......
driver/src/conv_driver.cpp
View file @ 58584c29
...@@ -30,20 +30,51 @@ int main(int argc, char* argv[]) ...@@ -30,20 +30,51 @@ int main(int argc, char* argv[])
using namespace ck; using namespace ck;
#if 0 #if 0
constexpr index_t N = 8; // 1x1
constexpr index_t C = 32; constexpr index_t N = 256;
constexpr index_t HI = 28; constexpr index_t C = 1024;
constexpr index_t WI = 28; constexpr index_t HI = 8;
constexpr index_t K = 32; constexpr index_t WI = 8;
constexpr index_t Y = 5; constexpr index_t K = 1024;
constexpr index_t X = 5; constexpr index_t Y = 1;
constexpr index_t X = 1;
using ConvStrides = Sequence<1, 1>; using ConvStrides = Sequence<1, 1>;
using ConvDilations = Sequence<2, 2>; using ConvDilations = Sequence<1, 1>;
using LeftPads = Sequence<0, 0>; using LeftPads = Sequence<0, 0>;
using RightPads = Sequence<0, 0>; using RightPads = Sequence<0, 0>;
#elif 0
// 1x7
constexpr index_t N = 128;
constexpr index_t C = 1024;
constexpr index_t HI = 17;
constexpr index_t WI = 17;
constexpr index_t K = 1024;
constexpr index_t Y = 1;
constexpr index_t X = 7;
using ConvStrides = Sequence<1, 1>;
using ConvDilations = Sequence<1, 1>;
using LeftPads = Sequence<0, 3>;
using RightPads = Sequence<0, 3>;
#elif 1 #elif 1
// 3x3, 34x34
constexpr index_t N = 64;
constexpr index_t C = 256;
constexpr index_t HI = 34;
constexpr index_t WI = 34;
constexpr index_t K = 256;
constexpr index_t Y = 3;
constexpr index_t X = 3;
using ConvStrides = Sequence<1, 1>;
using ConvDilations = Sequence<1, 1>;
using LeftPads = Sequence<0, 0>;
using RightPads = Sequence<0, 0>;
#elif 0
// 3x3, 34x34 // 3x3, 34x34
constexpr index_t N = 64; constexpr index_t N = 64;
constexpr index_t C = 256; constexpr index_t C = 256;
......
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