Skipped and commented out gpt neo tests

tenstorrent · Mar 4, 2025 · e25e9cf · e25e9cf
1 parent aaa26ad
commit e25e9cf
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 11 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
@@ -5,7 +5,6 @@
 from typing import Callable
 
 import pytest
-from infra import ModelTester, RunMode
 from utils import record_model_test_properties, runtime_fail
 
 from ..tester import GPTNeoTester
@@ -18,19 +17,21 @@
 
 @pytest.fixture
 def inference_tester() -> GPTNeoTester:
-    return GPTNeoTester(MODEL_PATH)
+    # return GPTNeoTester(MODEL_PATH)
+    return None
 
 
 @pytest.fixture
 def training_tester() -> GPTNeoTester:
-    return GPTNeoTester(ModelTester, run_mode=RunMode.TRAINING)
+    # return GPTNeoTester(ModelTester, run_mode=RunMode.TRAINING)
+    return None
 
 
 # ----- 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
@@ -5,7 +5,6 @@
 from typing import Callable
 
 import pytest
-from infra import ModelTester, RunMode
 from utils import record_model_test_properties, runtime_fail
 
 from ..tester import GPTNeoTester
@@ -18,19 +17,21 @@
 
 @pytest.fixture
 def inference_tester() -> GPTNeoTester:
-    return GPTNeoTester(MODEL_PATH)
+    # return GPTNeoTester(MODEL_PATH)
+    return None
 
 
 @pytest.fixture
 def training_tester() -> GPTNeoTester:
-    return GPTNeoTester(ModelTester, run_mode=RunMode.TRAINING)
+    # return GPTNeoTester(ModelTester, run_mode=RunMode.TRAINING)
+    return None
 
 
 # ----- 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
@@ -5,7 +5,6 @@
 from typing import Callable
 
 import pytest
-from infra import ModelTester, RunMode
 from utils import record_model_test_properties
 
 from ..tester import GPTNeoTester
@@ -18,12 +17,14 @@
 
 @pytest.fixture
 def inference_tester() -> GPTNeoTester:
-    return GPTNeoTester(MODEL_PATH)
+    # return GPTNeoTester(MODEL_PATH)
+    return None
 
 
 @pytest.fixture
 def training_tester() -> GPTNeoTester:
-    return GPTNeoTester(ModelTester, run_mode=RunMode.TRAINING)
+    # return GPTNeoTester(ModelTester, run_mode=RunMode.TRAINING)
+    return None
 
 
 # ----- Tests -----