diff --git a/src/opr-mm/test/io_remote.cpp b/src/opr-mm/test/io_remote.cpp index a68ffd9a53f700edd7d1b26700a70e078198f333..3f965f83af2dfc6a34842f0377791d527631770a 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();