Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 6 additions & 3 deletions .azure-pipelines/ut-basic-no-cover.yml
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,7 @@ stages:
dependsOn: []
jobs:
- job:
displayName: Test FWKs adaptor
steps:
- template: template/ut-template.yml
parameters:
Expand All @@ -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:
Expand All @@ -62,7 +64,7 @@ stages:
dependsOn: []
jobs:
- job:
displayName: PyTorch Pruning
displayName: Test PyTorch Pruning
steps:
- template: template/ut-template.yml
parameters:
Expand All @@ -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:
Expand All @@ -89,6 +91,7 @@ stages:
dependsOn: []
jobs:
- job:
displayName: Test other basic case
steps:
- template: template/ut-template.yml
parameters:
Expand Down
26 changes: 18 additions & 8 deletions .azure-pipelines/ut-basic.yml
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,7 @@ stages:
dependsOn: []
jobs:
- job:
displayName: Test FWKs adaptor
steps:
- template: template/ut-template.yml
parameters:
Expand All @@ -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:
Expand All @@ -58,7 +60,7 @@ stages:
dependsOn: []
jobs:
- job:
displayName: PyTorch Pruning
displayName: Test PyTorch Pruning
steps:
- template: template/ut-template.yml
parameters:
Expand All @@ -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:
Expand All @@ -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:
Expand All @@ -94,6 +97,7 @@ stages:
dependsOn: []
jobs:
- job:
displayName: Test ITEX
steps:
- template: template/ut-template.yml
parameters:
Expand All @@ -107,6 +111,7 @@ stages:
dependsOn: []
jobs:
- job:
displayName: Test other basic case
steps:
- template: template/ut-template.yml
parameters:
Expand All @@ -120,6 +125,7 @@ stages:
dependsOn: []
jobs:
- job:
displayName: Test FWKs adaptor baseline
steps:
- template: template/ut-template.yml
parameters:
Expand All @@ -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:
Expand All @@ -148,7 +155,7 @@ stages:
dependsOn: []
jobs:
- job:
displayName: PyTorch Pruning baseline
displayName: Test PyTorch Pruning baseline
steps:
- template: template/ut-template.yml
parameters:
Expand All @@ -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:
Expand All @@ -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:
Expand All @@ -187,6 +195,7 @@ stages:
dependsOn: []
jobs:
- job:
displayName: Test ITEX baseline
steps:
- template: template/ut-template.yml
parameters:
Expand All @@ -201,6 +210,7 @@ stages:
dependsOn: []
jobs:
- job:
displayName: Test other cases baseline
steps:
- template: template/ut-template.yml
parameters:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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
2 changes: 1 addition & 1 deletion neural_compressor/benchmark.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down