提交 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) { ...@@ -46,6 +46,7 @@ TEST(TestOprIORemote, Identity) {
} }
TEST(TestOprIORemote, IdentityMultiThread) { TEST(TestOprIORemote, IdentityMultiThread) {
REQUIRE_GPU(2);
auto cns = load_multiple_xpus(2); auto cns = load_multiple_xpus(2);
HostTensorGenerator<> gen; HostTensorGenerator<> gen;
auto host_x = gen({2, 3}, cns[1]); auto host_x = gen({2, 3}, cns[1]);
...@@ -79,6 +80,7 @@ TEST(TestOprIORemote, IdentityMultiThread) { ...@@ -79,6 +80,7 @@ TEST(TestOprIORemote, IdentityMultiThread) {
} }
TEST(TestOprIORemote, IdentityWithGopt) { TEST(TestOprIORemote, IdentityWithGopt) {
REQUIRE_GPU(2);
auto cns = load_multiple_xpus(2); auto cns = load_multiple_xpus(2);
HostTensorGenerator<> gen; HostTensorGenerator<> gen;
auto host_x = gen({2, 3}, cns[1]); auto host_x = gen({2, 3}, cns[1]);
...@@ -113,6 +115,7 @@ TEST(TestOprIORemote, IdentityWithGopt) { ...@@ -113,6 +115,7 @@ TEST(TestOprIORemote, IdentityWithGopt) {
} }
TEST(TestOprIORemote, APlusB) { TEST(TestOprIORemote, APlusB) {
REQUIRE_GPU(2);
auto cns = load_multiple_xpus(2); auto cns = load_multiple_xpus(2);
HostTensorGenerator<> gen; HostTensorGenerator<> gen;
auto host_x = gen({5, 7}, cns[0]), host_y = gen({5, 1}, cns[0]); auto host_x = gen({5, 7}, cns[0]), host_y = gen({5, 1}, cns[0]);
...@@ -164,6 +167,7 @@ TEST(TestOprIORemote, APlusB) { ...@@ -164,6 +167,7 @@ TEST(TestOprIORemote, APlusB) {
} }
TEST(TestOprIORemote, SendGrad) { TEST(TestOprIORemote, SendGrad) {
REQUIRE_GPU(2);
auto cns = load_multiple_xpus(2); auto cns = load_multiple_xpus(2);
HostTensorGenerator<> gen; HostTensorGenerator<> gen;
auto host_x = gen({2, 3}, cns[0]); 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.
先完成此消息的编辑!
想要评论请 注册