未验证 提交 6ac7cabe 编写于 作者: R Roc 提交者: GitHub

skip cpu test for test_index_select (#51741)

上级 ca364e14
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册