Create the shape correctly in python printout (#1520)

* Create the shape correctly in python printout

Create the shape correctly in python printout (#1520)
* Create the shape correctly in python printout
635502be · Paul Fultz II · GitHub · 72eab37d · 635502be
Unverified Commit 635502be authored Jan 26, 2023 by Paul Fultz II Committed by GitHub Jan 26, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/module.cpp src/module.cpp +2 -1

No files found.
--- a/src/module.cpp
+++ b/src/module.cpp
@@ -822,7 +822,8 @@ static void print_make_op(std::ostream& os, const operation& op)

 static void print_py_shape(std::ostream& os, const migraphx::shape& s)
 {
-    os << "migraphx.shape(" << s.type_string() << ", lens=" << to_json_string(s.lens());
+    os << "migraphx.shape(type=" << to_json_string(s.type_string())
+       << ", lens=" << to_json_string(s.lens());
    if(not s.standard())
        os << ", strides=" << to_json_string(s.strides());
    os << ")";