github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

split chain with parentheses for and node

Browse Source
This commit is contained in:
Szymon Ożóg
2023-08-23 09:43:41 +02:00
parent 9e96911771
commit 7499a7004e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/shape/symbolic.py
View File

@@ -287,5 +287,5 @@ render_python: Dict[Type, Callable] = {
ModNode: lambda self,ops,ctx: f"({self.a.render(ops,ctx)}%{self.b})",
LtNode: lambda self,ops,ctx: f"({self.a.render(ops,ctx)}<{sym_render(self.b,ops,ctx)})",
SumNode: lambda self,ops,ctx: f"({'+'.join(sorted([x.render(ops,ctx) for x in self.nodes]))})",
AndNode: lambda self,ops,ctx: f"({' and '.join(sorted([x.render(ops,ctx) for x in self.nodes]))})"
AndNode: lambda self,ops,ctx: ' and '.join(sorted([f"({x.render(ops, ctx)}" for x in self.nodes])) + len(self.nodes) * ')'
}