#8895: Fix dump_tensor(...) API to provide default arg

tenstorrent · Jun 1, 2024 · 3237870 · 3237870
1 parent 66207ac
commit 3237870
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tt_eager/tt_lib/csrc/tt_lib_bindings_tensor.cpp b/tt_eager/tt_lib/csrc/tt_lib_bindings_tensor.cpp
@@ -922,6 +922,9 @@ void TensorModule(py::module& m_tensor) {
     m_tensor.def(
         "dump_tensor",
         &dump_tensor,
+        py::arg("filename"),
+        py::arg("tensor"),
+        py::arg("strategy") = std::unordered_map<std::string, std::string>{},
         R"doc(
             Dump tensor to file
         )doc");