fix ptq-hpo on the windows platform (#1032)

* fix install

* update
Co-authored-by: NGuanghua Yu <742925032@qq.com>

paddleslim/auto_compression/compressor.py

@@ -18,9 +18,11 @@ import sys
 import numpy as np
 import inspect
 from collections import namedtuple, Iterable
+import platform
 import paddle
 import paddle.distributed.fleet as fleet
-from ..quant.quant_post_hpo import quant_post_hpo
+if platform.system().lower() == 'linux':
+    from ..quant.quant_post_hpo import quant_post_hpo
 from ..quant.quanter import convert
 from ..common.recover_program import recover_inference_program
 from ..common import get_logger
@@ -231,6 +233,10 @@ class AutoCompression:
     def compress(self):
         ### start compress, including train/eval model
         if self._strategy == 'ptq_hpo':
+            if platform.system().lower() != 'linux':
+                raise NotImplementedError(
+                    "post-quant-hpo is not support in system other than linux")
+
             quant_post_hpo(
                 self._exe,
                 self._places,

paddleslim/quant/__init__.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import platform
 import logging
 
 import paddle
@@ -32,7 +33,12 @@ try:
     from .quanter import quant_aware, convert, quant_post_static, quant_post_dynamic
     from .quanter import quant_post, quant_post_only_weight
     from .quant_aware_with_infermodel import quant_aware_with_infermodel, export_quant_infermodel
-    from .quant_post_hpo import quant_post_hpo
+    if platform.system().lower() == 'linux':
+        from .quant_post_hpo import quant_post_hpo
+    else:
+        _logger.warning(
+            "post-quant-hpo is not support in system other than linux")
+
 except Exception as e:
     _logger.warning(e)
     _logger.warning(

requirements.txt

@@ -7,4 +7,4 @@ pillow
 pyyaml
 scikit-learn
 smac
-paddleslim-opt-tools
+paddleslim-opt-tools
\ No newline at end of file