Fix Windows multi-instance bug (#216)

Signed-off-by: spycsh <[email protected]>
Co-authored-by: chensuyue <[email protected]>

Author: Spycsh

examples/.config/model_params_onnxrt_win.json

@@ -7,16 +7,16 @@
             "yaml": "resnet50_v1_5.yaml",
             "strategy": "basic",
             "batch_size": 100,
-            "new_benchmark": false
+            "new_benchmark": true
         },
         "bert_base_MRPC_static": {
-            "model_src_dir": "language_translation/bert/quantization/ptq",
+            "model_src_dir": "nlp/bert/quantization/ptq",
             "dataset_location": "glue_data/MRPC",
             "input_model": "models/onnx/bert.onnx",
             "yaml": "bert_static.yaml",
             "strategy": "basic",
             "batch_size": 8,
-            "new_benchmark": false
+            "new_benchmark": true
         },
         "mobilenet_v2": {
             "model_src_dir": "image_recognition/mobilenet_v2/quantization/ptq",
@@ -25,7 +25,7 @@
             "yaml": "mobilenet_v2.yaml",
             "strategy": "basic",
             "batch_size": 100,
-            "new_benchmark": false
+            "new_benchmark": true
         }
     }
 }

neural_compressor/experimental/benchmark.py

@@ -225,13 +225,13 @@ def config_instance(self):
         num_of_instance = int(os.environ.get('NUM_OF_INSTANCE'))
         cores_per_instance = int(os.environ.get('CORES_PER_INSTANCE'))
 
-        if(get_architecture() == 'aarch64' and int(get_threads_per_core()) > 1):
+        if(sys.platform in ['linux'] and get_architecture() == 'aarch64' and int(get_threads_per_core()) > 1):
             raise OSError('Currently no support on AMD with hyperthreads')
-        else:
+        elif sys.platform in ['linux']:
             bounded_threads = get_bounded_threads(get_core_ids(), get_threads(), get_physical_ids())
 
         for i in range(0, num_of_instance):
-            if get_architecture() == 'x86_64':
+            if sys.platform in ['linux'] and get_architecture() == 'x86_64':
                 core_list_idx = np.arange(0, cores_per_instance) + i * cores_per_instance
                 core_list = np.array(bounded_threads)[core_list_idx]
             else: