fix bug in ZPull_

niehao100 · niehao100 · commit b33132ea0d44 · 2025-11-03T20:03:13.000+08:00
diff --git a/include/ps/af_tensor_app.h b/include/ps/af_tensor_app.h
@@ -294,7 +294,7 @@ class AFTensorWorker {
       msg.meta.dtype = static_cast<int>(tensor.scalar_type());
       msg.meta.shape.clear();
       for (int64_t s = 0; s < tensor.dim(); s++) {
-        msg.meta.shape.push_back(tensor.size(i));
+        msg.meta.shape.push_back(tensor.size(s));
       }
       msg.data.clear();
       msg.AddData(key);

Original file line number	Diff line number	Diff line change
`@@ -294,7 +294,7 @@ class AFTensorWorker {`
`294`	`294`	`msg.meta.dtype = static_cast<int>(tensor.scalar_type());`
`295`	`295`	`msg.meta.shape.clear();`
`296`	`296`	`for (int64_t s = 0; s < tensor.dim(); s++) {`
`297`		`- msg.meta.shape.push_back(tensor.size(i));`
	`297`	`+ msg.meta.shape.push_back(tensor.size(s));`
`298`	`298`	`}`
`299`	`299`	`msg.data.clear();`
`300`	`300`	`msg.AddData(key);`