Skip to content

Commit 7014eac

Browse files
committed
#783: Add inf=float(inf) in generate_op_accuracy_tests.py
1 parent 439f5ba commit 7014eac

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tools/generate_op_accuracy_tests.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -289,6 +289,7 @@ def _build_code_from_aten_ttnn_graphs(aten_graph, ttnn_graph, output_nodes):
289289
# comment out signature if not the first graph
290290
graph_code = [forward_signature] if len(output_nodes) == 0 else [" # " + forward_signature]
291291
graph_code.append(" device = ttnn.open_device(device_id=0, l1_small_size=16384)")
292+
graph_code.append(""" inf = float("inf")""")
292293
for node in aten_all_nodes:
293294
if node.op == "output":
294295
output_nodes.append(node.args[0])

0 commit comments

Comments
 (0)