diff --git a/.azure-pipelines/ut-basic-no-cover.yml b/.azure-pipelines/ut-basic-no-cover.yml index 1bd4c3574d3..2f2e29e1148 100644 --- a/.azure-pipelines/ut-basic-no-cover.yml +++ b/.azure-pipelines/ut-basic-no-cover.yml @@ -32,6 +32,7 @@ stages: dependsOn: [] jobs: - job: + displayName: Test FWKs adaptor steps: - template: template/ut-template.yml parameters: @@ -43,10 +44,11 @@ stages: utContainerName: "utTest-no-coverage" - stage: API - displayName: User facing API + displayName: Unit Test User facing API dependsOn: [] jobs: - job: + displayName: Test User facing API steps: - template: template/ut-template.yml parameters: @@ -62,7 +64,7 @@ stages: dependsOn: [] jobs: - job: - displayName: PyTorch Pruning + displayName: Test PyTorch Pruning steps: - template: template/ut-template.yml parameters: @@ -73,7 +75,7 @@ stages: utTestMode: "no-coverage" utContainerName: "utTest-no-coverage" - job: - displayName: TensorFlow Pruning + displayName: Test TensorFlow Pruning steps: - template: template/ut-template.yml parameters: @@ -89,6 +91,7 @@ stages: dependsOn: [] jobs: - job: + displayName: Test other basic case steps: - template: template/ut-template.yml parameters: diff --git a/.azure-pipelines/ut-basic.yml b/.azure-pipelines/ut-basic.yml index 64110e55fdc..3615a00cecb 100644 --- a/.azure-pipelines/ut-basic.yml +++ b/.azure-pipelines/ut-basic.yml @@ -32,6 +32,7 @@ stages: dependsOn: [] jobs: - job: + displayName: Test FWKs adaptor steps: - template: template/ut-template.yml parameters: @@ -41,10 +42,11 @@ stages: utArtifact: "ut-coverage-adaptor" - stage: API - displayName: User facing API + displayName: Unit Test User facing API dependsOn: [] jobs: - job: + displayName: Test User facing API steps: - template: template/ut-template.yml parameters: @@ -58,7 +60,7 @@ stages: dependsOn: [] jobs: - job: - displayName: PyTorch Pruning + displayName: Test PyTorch Pruning steps: - template: template/ut-template.yml parameters: @@ -67,7 +69,7 @@ stages: uploadPath: $(UPLOAD_PATH) utArtifact: "ut-coverage-pt-pruning" - job: - displayName: TensorFlow Pruning + displayName: Test TensorFlow Pruning steps: - template: template/ut-template.yml parameters: @@ -77,10 +79,11 @@ stages: utArtifact: "ut-coverage-tf-pruning" - stage: TFNewAPI - displayName: Unit Test tf newAPI + displayName: Unit Test TF newAPI dependsOn: [] jobs: - job: + displayName: Test TF newAPI steps: - template: template/ut-template.yml parameters: @@ -94,6 +97,7 @@ stages: dependsOn: [] jobs: - job: + displayName: Test ITEX steps: - template: template/ut-template.yml parameters: @@ -107,6 +111,7 @@ stages: dependsOn: [] jobs: - job: + displayName: Test other basic case steps: - template: template/ut-template.yml parameters: @@ -120,6 +125,7 @@ stages: dependsOn: [] jobs: - job: + displayName: Test FWKs adaptor baseline steps: - template: template/ut-template.yml parameters: @@ -130,10 +136,11 @@ stages: repo: $(REPO) - stage: API_base - displayName: User facing API baseline + displayName: Unit Test User facing API baseline dependsOn: [] jobs: - job: + displayName: Test User facing API baseline steps: - template: template/ut-template.yml parameters: @@ -148,7 +155,7 @@ stages: dependsOn: [] jobs: - job: - displayName: PyTorch Pruning baseline + displayName: Test PyTorch Pruning baseline steps: - template: template/ut-template.yml parameters: @@ -158,7 +165,7 @@ stages: utArtifact: "ut-coverage-pt-pruning-base" repo: $(REPO) - job: - displayName: TensorFlow Pruning baseline + displayName: Test TensorFlow Pruning baseline steps: - template: template/ut-template.yml parameters: @@ -169,10 +176,11 @@ stages: repo: $(REPO) - stage: TFNewAPI_base - displayName: Unit Test tf newAPI baseline + displayName: Unit Test TF newAPI baseline dependsOn: [] jobs: - job: + displayName: Test TF newAPI baseline steps: - template: template/ut-template.yml parameters: @@ -187,6 +195,7 @@ stages: dependsOn: [] jobs: - job: + displayName: Test ITEX baseline steps: - template: template/ut-template.yml parameters: @@ -201,6 +210,7 @@ stages: dependsOn: [] jobs: - job: + displayName: Test other cases baseline steps: - template: template/ut-template.yml parameters: diff --git a/examples/tensorflow/nlp/distilbert_base/quantization/ptq/requirements.txt b/examples/tensorflow/nlp/distilbert_base/quantization/ptq/requirements.txt index 2a3842c85d2..e60cdb323dd 100644 --- a/examples/tensorflow/nlp/distilbert_base/quantization/ptq/requirements.txt +++ b/examples/tensorflow/nlp/distilbert_base/quantization/ptq/requirements.txt @@ -3,5 +3,5 @@ scikit-learn>=1.1.2 scipy>=1.9.3 sklearn==0.0 tokenizers==0.13.1 -transformers==4.23.1 +transformers>=4.31.0 intel-tensorflow>=2.12.0 diff --git a/neural_compressor/benchmark.py b/neural_compressor/benchmark.py index fdc136d5172..d370519278e 100644 --- a/neural_compressor/benchmark.py +++ b/neural_compressor/benchmark.py @@ -475,7 +475,7 @@ def benchmark_with_raw_cmd(raw_cmd, conf=None): if conf.backend == "ipex": import intel_extension_for_pytorch assert sys.platform in ["linux", "win32"], "only support platform windows and linux..." - # disable multi-instance for running bechmark on GPU device + # disable multi-instance for running benchmark on GPU device set_all_env_var(conf) config_instance(raw_cmd)