Cleanup test_performance.py [perf]

ragulpr · Jan 19, 2025 · fbb2a95 · fbb2a95
1 parent 173f350
commit fbb2a95
Showing 1 changed file with 0 additions and 11 deletions.
diff --git a/test_performance.py b/test_performance.py
@@ -24,17 +24,6 @@
 print(f'torch.__version__: {torch.__version__}')
 print(args)
 
-import time
-import math
-
-import os
-os.environ['OMP_NUM_THREADS'] = '16'
-os.environ['NUMEXPR_MAX_THREADS'] = '16'
-
-# Configure PyTorch threading
-torch.set_num_threads(16)  # Intra-op threads
-torch.set_num_interop_threads(16)  # Inter-op threads
-
 def time_since(since):
     s = time.time() - since
     m = math.floor(s / 60)