提交 4cf193ab 编写于 作者: M Megvii Engine Team

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

GitOrigin-RevId: 0184036b8bc35165b97b35155960a0ef7756e5d9
上级 c76789a5
......@@ -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<test::MockGroupClient>();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册