From 4cf193ab97a0fb1eda64459d0aaa04f216c2f76b Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 21 Jul 2020 10:23:17 +0800 Subject: [PATCH] fix(mgb/opr-mm): fix device id in TestOprIORemote.IdentityWithGopt GitOrigin-RevId: 0184036b8bc35165b97b35155960a0ef7756e5d9 --- src/opr-mm/test/io_remote.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/opr-mm/test/io_remote.cpp b/src/opr-mm/test/io_remote.cpp index a68ffd9a..3f965f83 100644 --- a/src/opr-mm/test/io_remote.cpp +++ b/src/opr-mm/test/io_remote.cpp @@ -81,7 +81,7 @@ TEST(TestOprIORemote, IdentityMultiThread) { TEST(TestOprIORemote, IdentityWithGopt) { auto cns = load_multiple_xpus(2); HostTensorGenerator<> gen; - auto host_x = gen({2, 3}, cns[0]); + auto host_x = gen({2, 3}, cns[1]); HostTensorND host_x_get; auto client = std::make_shared(); -- GitLab