提交 6af0299c 编写于 作者: M Megvii Engine Team

test(mge/ioremote): add REQUIRE_GPU for ioremote opr test

GitOrigin-RevId: 31ec8982c4cc59839d1ac0060ec7c774e6f04dff
上级 bab0caab
......@@ -46,6 +46,7 @@ TEST(TestOprIORemote, Identity) {
}
TEST(TestOprIORemote, IdentityMultiThread) {
REQUIRE_GPU(2);
auto cns = load_multiple_xpus(2);
HostTensorGenerator<> gen;
auto host_x = gen({2, 3}, cns[1]);
......@@ -79,6 +80,7 @@ TEST(TestOprIORemote, IdentityMultiThread) {
}
TEST(TestOprIORemote, IdentityWithGopt) {
REQUIRE_GPU(2);
auto cns = load_multiple_xpus(2);
HostTensorGenerator<> gen;
auto host_x = gen({2, 3}, cns[1]);
......@@ -113,6 +115,7 @@ TEST(TestOprIORemote, IdentityWithGopt) {
}
TEST(TestOprIORemote, APlusB) {
REQUIRE_GPU(2);
auto cns = load_multiple_xpus(2);
HostTensorGenerator<> gen;
auto host_x = gen({5, 7}, cns[0]), host_y = gen({5, 1}, cns[0]);
......@@ -164,6 +167,7 @@ TEST(TestOprIORemote, APlusB) {
}
TEST(TestOprIORemote, SendGrad) {
REQUIRE_GPU(2);
auto cns = load_multiple_xpus(2);
HostTensorGenerator<> gen;
auto host_x = gen({2, 3}, cns[0]);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册