No changes to utils.py

tenstorrent · Mar 1, 2025 · 9e5dae8 · 9e5dae8
1 parent 237ca12
commit 9e5dae8
Showing 1 changed file with 0 additions and 4 deletions.
diff --git 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)