Skip to content

Commit

Permalink
Skipping additional set of model tests due to the CI limitations (#1047)
Browse files Browse the repository at this point in the history
Main reasons:
- MLIR seg faults (not CI related, but causes seg faults)
- Insuficient DRAM space
  • Loading branch information
nvukobratTT authored Jan 15, 2025
1 parent fc4b243 commit 773304e
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 0 deletions.
2 changes: 2 additions & 0 deletions forge/test/models/pytorch/text/phi3/test_phi3.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,8 @@
@pytest.mark.nightly
@pytest.mark.parametrize("variant", variants)
def test_phi3_causal_lm(record_forge_property, variant):
pytest.skip("Insufficient host DRAM to run this model (requires a bit more than 64 GB)")

# Build Module Name
module_name = build_module_name(framework=Framework.PYTORCH, model="phi3", variant=variant, task=Task.CAUSAL_LM)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,8 @@ def generate_model_mobilenetV2I96_imgcls_hf_pytorch(variant):
@pytest.mark.nightly
@pytest.mark.parametrize("variant", ["google/mobilenet_v2_0.35_96"])
def test_mobilenetv2_96(record_forge_property, variant):
pytest.skip("Hitting segmentation fault in MLIR")

# Build Module Name
module_name = build_module_name(
framework=Framework.PYTORCH, model="mobilenetv2", variant=variant, source=Source.HUGGINGFACE
Expand Down Expand Up @@ -106,6 +108,8 @@ def generate_model_mobilenetV2I160_imgcls_hf_pytorch(variant):
@pytest.mark.nightly
@pytest.mark.parametrize("variant", ["google/mobilenet_v2_0.75_160"])
def test_mobilenetv2_160(record_forge_property, variant):
pytest.skip("Hitting segmentation fault in MLIR")

# Build Module Name
module_name = build_module_name(
framework=Framework.PYTORCH, model="mobilenetv2", variant=variant, source=Source.HUGGINGFACE
Expand Down Expand Up @@ -141,6 +145,8 @@ def generate_model_mobilenetV2I244_imgcls_hf_pytorch(variant):
@pytest.mark.nightly
@pytest.mark.parametrize("variant", ["google/mobilenet_v2_1.0_224"])
def test_mobilenetv2_224(record_forge_property, variant):
pytest.skip("Hitting segmentation fault in MLIR")

# Build Module Name
module_name = build_module_name(
framework=Framework.PYTORCH, model="mobilenetv2", variant=variant, source=Source.HUGGINGFACE
Expand Down

0 comments on commit 773304e

Please sign in to comment.