From 6af0299c3f6171565109591f79ff0798e877cb2c Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Thu, 1 Apr 2021 14:31:20 +0800
Subject: [PATCH] test(mge/ioremote): add REQUIRE_GPU for ioremote opr test

GitOrigin-RevId: 31ec8982c4cc59839d1ac0060ec7c774e6f04dff
---
 src/opr-mm/test/io_remote.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/opr-mm/test/io_remote.cpp b/src/opr-mm/test/io_remote.cpp
index e58cbe32c..95668f258 100644
--- a/src/opr-mm/test/io_remote.cpp
+++ b/src/opr-mm/test/io_remote.cpp
@@ -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]);
-- 
GitLab