Fix Convert IR (#189)

* nit * add docs * fix memcpy annotation; fix conv2d ir; fix frontend convert * code style * add check symmetric padding * Fix Convert IR Co-authored-by: yuqing <[email protected]>
microsoft · Dec 22, 2020 · 392811e · 392811e
1 parent 61418e9
commit 392811e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/nnfusion/core/operators/generic_op/generic_op_define/Convert.cpp b/src/nnfusion/core/operators/generic_op/generic_op_define/Convert.cpp
@@ -39,7 +39,9 @@ REGISTER_OP(Convert)
                                     << op->get_convert_element_type().c_type_string();
 
         return op::create_code_from_template(
-            "@output0@@data_layout@ = @input0@@[email protected](\"@out_dtype@\");",
-            {{"data_layout", op::create_layout_from_dims(gnode->get_output_shape(0))},
+            "@output0@@data_layout@ = @input0@@[email protected](`@out_dtype@`);",
+            {{"data_layout",
+              vector_to_string<std::vector<std::string>>(
+                  op::create_layout_from_dims(gnode->get_output_shape(0)))},
              {"out_dtype", out_dtype}});
     });