Browse Source

fix(mgb/opr-mm): fix device id in TestOprIORemote.IdentityWithGopt

GitOrigin-RevId: 0184036b8b
release-0.6
Megvii Engine Team 4 years ago
parent
commit
4cf193ab97
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      src/opr-mm/test/io_remote.cpp

+ 1
- 1
src/opr-mm/test/io_remote.cpp View File

@@ -81,7 +81,7 @@ TEST(TestOprIORemote, IdentityMultiThread) {
TEST(TestOprIORemote, IdentityWithGopt) { TEST(TestOprIORemote, IdentityWithGopt) {
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[1]);
HostTensorND host_x_get; HostTensorND host_x_get;
auto client = std::make_shared<test::MockGroupClient>(); auto client = std::make_shared<test::MockGroupClient>();




Loading…
Cancel
Save