From aee5840951c7cf9947eb3759533bdcead05ac5a5 Mon Sep 17 00:00:00 2001 From: Austin Ho Date: Wed, 29 May 2024 15:02:44 +0000 Subject: [PATCH] Revert "#0: Run resnet E2E in LoFi" This reverts commit 486f3808507e892bc3456bbd05a70d9ba38f8231. --- models/demos/resnet/tests/test_perf_resnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/demos/resnet/tests/test_perf_resnet.py b/models/demos/resnet/tests/test_perf_resnet.py index 8d37af57b75..f8811cd01cc 100644 --- a/models/demos/resnet/tests/test_perf_resnet.py +++ b/models/demos/resnet/tests/test_perf_resnet.py @@ -18,7 +18,7 @@ from models.demos.resnet.tt.metalResnetBlock50 import ResNet, Bottleneck model_config = { - "MATH_FIDELITY": tt_lib.tensor.MathFidelity.LoFi, + "MATH_FIDELITY": tt_lib.tensor.MathFidelity.HiFi2, "WEIGHTS_DTYPE": tt_lib.tensor.DataType.BFLOAT8_B, "ACTIVATIONS_DTYPE": tt_lib.tensor.DataType.BFLOAT8_B, }