diff --git a/python/paddle/fluid/tests/unittests/test_index_select_op.py b/python/paddle/fluid/tests/unittests/test_index_select_op.py index f318cfab10a14730a7552d30c939f842d8ba1f85..7fe064af6cbf9aaee7201a97d622b049c31f33f4 100644 --- a/python/paddle/fluid/tests/unittests/test_index_select_op.py +++ b/python/paddle/fluid/tests/unittests/test_index_select_op.py @@ -95,6 +95,9 @@ class TestIndexSelectFP16OP(TestIndexSelectOp): # no scatter op (the backward op of index_select/gather) for bf16 +@unittest.skipIf( + not paddle.is_compiled_with_cuda(), "paddle is not compiled with cuda" +) class TestIndexSelectBF16Op(OpTest): def setUp(self): self.python_api = paddle.index_select