Skip to content

Commit a9d9f98

Browse files
committed
hotfix: those tests fail locally on mac due to buffer count
1 parent 2454bf0 commit a9d9f98

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

Diff for: test/models/test_train.py

+2
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ def test_convnext(self):
4040
check_gc()
4141

4242
@unittest.skipIf(CI, "slow")
43+
@unittest.skipIf(Device.DEFAULT in ["METAL", "WEBGPU"], "too many buffers for webgpu and metal")
4344
def test_efficientnet(self):
4445
model = EfficientNet(0)
4546
X = np.zeros((BS,3,224,224), dtype=np.float32)
@@ -56,6 +57,7 @@ def test_vit(self):
5657
train_one_step(model,X,Y)
5758
check_gc()
5859

60+
@unittest.skipIf(Device.DEFAULT in ["METAL", "WEBGPU"], "too many buffers for webgpu and metal")
5961
def test_transformer(self):
6062
# this should be small GPT-2, but the param count is wrong
6163
# (real ff_dim is 768*4)

0 commit comments

Comments
 (0)