From 79250c05e9f885d9ee961847ec443921ad2ba8cc Mon Sep 17 00:00:00 2001 From: Kristijan Mitrovic Date: Tue, 4 Mar 2025 18:09:05 +0000 Subject: [PATCH] Skipped gpt neo tests --- tests/jax/models/gpt_neo/gpt_neo_125m/test_gpt_neo_125m.py | 6 +++--- tests/jax/models/gpt_neo/gpt_neo_1_3b/test_gpt_neo_1_3b.py | 6 +++--- tests/jax/models/gpt_neo/gpt_neo_2_7b/test_gpt_neo_2_7b.py | 4 ++-- 3 files changed, 8 insertions(+), 8 deletions(-) diff --git a/tests/jax/models/gpt_neo/gpt_neo_125m/test_gpt_neo_125m.py b/tests/jax/models/gpt_neo/gpt_neo_125m/test_gpt_neo_125m.py index fceb6a0b..1da5cb50 100644 --- a/tests/jax/models/gpt_neo/gpt_neo_125m/test_gpt_neo_125m.py +++ b/tests/jax/models/gpt_neo/gpt_neo_125m/test_gpt_neo_125m.py @@ -5,7 +5,7 @@ from typing import Callable import pytest -from infra import ModelTester, RunMode +from infra import RunMode from utils import record_model_test_properties, runtime_fail from ..tester import GPTNeoTester @@ -23,14 +23,14 @@ def inference_tester() -> GPTNeoTester: @pytest.fixture def training_tester() -> GPTNeoTester: - return GPTNeoTester(ModelTester, run_mode=RunMode.TRAINING) + return GPTNeoTester(MODEL_PATH, run_mode=RunMode.TRAINING) # ----- Tests ----- @pytest.mark.nightly -@pytest.mark.xfail( +@pytest.mark.skip( reason=runtime_fail( "Host data with total size 4B does not match expected size 2B of device buffer!" ) diff --git a/tests/jax/models/gpt_neo/gpt_neo_1_3b/test_gpt_neo_1_3b.py b/tests/jax/models/gpt_neo/gpt_neo_1_3b/test_gpt_neo_1_3b.py index 4c77d8a1..e158bbe9 100644 --- a/tests/jax/models/gpt_neo/gpt_neo_1_3b/test_gpt_neo_1_3b.py +++ b/tests/jax/models/gpt_neo/gpt_neo_1_3b/test_gpt_neo_1_3b.py @@ -5,7 +5,7 @@ from typing import Callable import pytest -from infra import ModelTester, RunMode +from infra import RunMode from utils import record_model_test_properties, runtime_fail from ..tester import GPTNeoTester @@ -23,14 +23,14 @@ def inference_tester() -> GPTNeoTester: @pytest.fixture def training_tester() -> GPTNeoTester: - return GPTNeoTester(ModelTester, run_mode=RunMode.TRAINING) + return GPTNeoTester(MODEL_PATH, run_mode=RunMode.TRAINING) # ----- Tests ----- @pytest.mark.nightly -@pytest.mark.xfail( +@pytest.mark.skip( reason=runtime_fail( "Host data with total size 4B does not match expected size 2B of device buffer!" ) diff --git a/tests/jax/models/gpt_neo/gpt_neo_2_7b/test_gpt_neo_2_7b.py b/tests/jax/models/gpt_neo/gpt_neo_2_7b/test_gpt_neo_2_7b.py index 85c475eb..0e36ae25 100644 --- a/tests/jax/models/gpt_neo/gpt_neo_2_7b/test_gpt_neo_2_7b.py +++ b/tests/jax/models/gpt_neo/gpt_neo_2_7b/test_gpt_neo_2_7b.py @@ -5,7 +5,7 @@ from typing import Callable import pytest -from infra import ModelTester, RunMode +from infra import RunMode from utils import record_model_test_properties from ..tester import GPTNeoTester @@ -23,7 +23,7 @@ def inference_tester() -> GPTNeoTester: @pytest.fixture def training_tester() -> GPTNeoTester: - return GPTNeoTester(ModelTester, run_mode=RunMode.TRAINING) + return GPTNeoTester(MODEL_PATH, run_mode=RunMode.TRAINING) # ----- Tests -----