mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages 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
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) * ')'
}