Nhwc fwd

driver/args.h

@@ -157,14 +157,37 @@ class args_t {
     std::unordered_map<char, args_input_t> input_map;
 };
 
+static inline std::string create_base_args(int argc, char *argv[]) {
+    if(argc < 2)
+    {
+        printf("Invalid Number of Input Arguments\n");
+        exit(0);
+    }
+
+    std::string arg = argv[1];
+
+    if(arg != "conv" && arg != "convfp16" && arg != "convint8" && arg != "--version")
+    {
+        printf("Invalid Base Input Argument\n");
+        exit(0);
+    }
+    else if(arg == "-h" || arg == "--help" || arg == "-?")
+        exit(0);
+    else
+        return arg;
+}
+
 static inline args_t create_conv_args(int argc, char *argv[]) {
-    const std::string base("conv");
+    const std::string base = create_base_args(argc, argv);
     if (argc >= 2 && argv[1] != base) {
         printf("not proper base arg name");
         exit(1);
     }
 
     args_t args;
+    args.insert_arg("in_layout", 'I', "NCHW", "Input Layout (Default=NCHW)", "string");
+    args.insert_arg("out_layout", 'O', "NCHW", "Output Layout (Default=NCHW)", "string");
+    args.insert_arg("fil_layout", 'f', "NCHW", "Input Layout (Default=NCHW)", "string");
     args.insert_arg("spatial_dim", '_', "2",
                     "convolution spatial dimension (Default-2)", "int");
     args.insert_arg("forw", 'F', "0", "Flag enables fwd, bwd, wrw convolutions"