Skip to content

Commit 9c13878

Browse files
committed
Skip GPU failing test
1 parent c700590 commit 9c13878

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/plugins/test_custom_plugin.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,9 @@
1111
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
14+
import pytest
15+
import torch
16+
1417
from pytorch_lightning import Trainer
1518
from pytorch_lightning.plugins import DDPPlugin
1619
from tests.helpers import BoringModel
@@ -26,6 +29,7 @@ def __init__(self, **kwargs):
2629

2730

2831
@RunIf(skip_windows=True)
32+
@pytest.mark.skipif(torch.cuda.is_available(), reason="RuntimeError: Tensors must be CUDA and dense")
2933
def test_sync_batchnorm_set(tmpdir):
3034
"""Tests if sync_batchnorm is automatically set for custom plugin."""
3135
model = BoringModel()

0 commit comments

Comments
 (0)