From 3a863ca8ddabf4d4921b2752e07bfcef4def4bd3 Mon Sep 17 00:00:00 2001
From: Daniel King <daniel@mosaicml.com>
Date: Fri, 17 Nov 2023 16:55:41 -0800
Subject: [PATCH] fix

---
 tests/test_model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test_model.py b/tests/test_model.py
index 4ed3a385cb..64de3a3cff 100644
--- a/tests/test_model.py
+++ b/tests/test_model.py
@@ -304,6 +304,7 @@ def test_full_forward_and_backward_t5_small(batch_size: int = 2):
     assert not torch.equal(original_params, updated_params)
 
 
+@pytest.mark.gpu
 @pytest.mark.parametrize(
     'attn_impl,precision',
     [('torch', torch.float16), ('torch', torch.bfloat16),