From 9e5dae8345072574e366ff2b21a8b388ae7d9e18 Mon Sep 17 00:00:00 2001 From: Kevin Wu Date: Sat, 1 Mar 2025 00:15:09 +0000 Subject: [PATCH] No changes to utils.py --- tests/utils.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/tests/utils.py b/tests/utils.py index 1656381b7..94397ccc3 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -3,7 +3,6 @@ import collections import re from typing import List, Dict, Tuple -from torch_ttnn.cpp_extension.custom_device_mode import ttnn_module, enable_ttnn_device class ModelTester: @@ -131,9 +130,6 @@ def test_model_eval(self, as_ttnn=False, option=None): model = self.set_model_eval(self.model) inputs = self.set_inputs_eval(self.inputs) if as_ttnn == True: - torch.utils.rename_privateuse1_backend("ttnn") - ttnn_device = ttnn_module.custom_device_from_ttnn(option.device) - inputs = inputs.to(ttnn_device) model = self.compile_model(model, option) outputs = self.run_model(model, inputs) results = self.get_results_eval(model, inputs, outputs)