diff --git a/doc/tutorial.rst b/doc/tutorial.rst index 110e35bbb76e08c922f11086de761ddfe0bbf8da..54b78daa176c8d29b59eb8a46be0a79ec28e2159 100644 --- a/doc/tutorial.rst +++ b/doc/tutorial.rst @@ -316,7 +316,7 @@ that these dependencies show up there, too: .. doctest:: - >>> print(knl) + >>> print(knl.stringify(with_dependencies=True)) --------------------------------------------------------------------------- KERNEL: loopy_kernel --------------------------------------------------------------------------- diff --git a/loopy/kernel/__init__.py b/loopy/kernel/__init__.py index 917279408c9fbb68fd8b75b7087157ab4b281cdb..e42aa1fd790008f5e0c6bae3847978993815679b 100644 --- a/loopy/kernel/__init__.py +++ b/loopy/kernel/__init__.py @@ -964,7 +964,7 @@ class LoopKernel(RecordWithoutPickling): # {{{ pretty-printing - def __str__(self): + def stringify(self, with_dependencies=False): lines = [] from loopy.preprocess import add_default_dependencies @@ -1066,8 +1066,9 @@ class LoopKernel(RecordWithoutPickling): if dep_lines: lines.append(sep) lines.append("DEPENDENCIES: " - "use loopy.show_dependency_graph to visualize") - #lines.extend(dep_lines) + "(use loopy.show_dependency_graph to visualize)") + if with_dependencies: + lines.extend(dep_lines) lines.append(sep) @@ -1079,6 +1080,9 @@ class LoopKernel(RecordWithoutPickling): return "\n".join(lines) + def __str__(self): + return self.stringify() + # }}} # {{{ implementation arguments