send/resp:通信增加了返回确认

Author: miaobyte

excuter/op-mem-ompsimd/src/client/main.cpp

@@ -20,17 +20,22 @@ int main()
 
     server.func = [&mem, &opfactory, &memmutex](const char *buffer)->std::string
     {
+
+        
         deepx::op::Op op;
         op.recv_at = chrono::system_clock::now();
         op.load(buffer);
+        std::string resp=to_string(op.id);
+        resp+=" recv_at:";
+        resp+=to_string(op.recv_at.time_since_epoch().count());
         if (opfactory.ops.find(op.name)==opfactory.ops.end()){
             cout<<"<op> "<<op.name<<" not found"<<endl;
-            return "error";
+            resp+="error op not found";
         }
         auto &type_map = opfactory.ops.find(op.name)->second;
         if (type_map.find(op.dtype)==type_map.end()){
             cout<<"<op>"<<op.name<<" "<<op.dtype<<" not found"<<endl;
-            return "error";
+            resp+="error dtype not found";
         }
         auto src = type_map.find(op.dtype)->second;
 
@@ -42,7 +47,8 @@ int main()
            (*src).forward(mem);
         }
         memmutex.unlock();
-        return to_string(op.id);
+        resp+=" success";
+        return resp;
     };
     server.start();
     return 0;

front/py/deepx/nn/deepxir.py

@@ -1,7 +1,6 @@
 from typing import Tuple, List, Optional
 import time
 
-_id_counter=0
 class DeepxIR:
     def __init__(self, 
                 name:str,
@@ -39,7 +38,7 @@ def __init__(self,
         self._grad = grad
         self._args_grad = args_grad if grad else []
         self._returns_grad = returns_grad if grad else []
-        self._id=++_id_counter
+        self._id=None
         self._created_at=time.time()
         self._sent_at=None
 

front/py/deepx/scheduler/client/allclient.py

@@ -4,6 +4,12 @@
 import time
 default_client = _default_udpconn
 
+
+_id_counter=0
 def send(ir:DeepxIR) -> Optional[dict]:
     ir._sent_at=time.time()
-    return default_client.send(str(ir))
+    global _id_counter
+    _id_counter=_id_counter+1
+    ir._id=_id_counter
+    resp=default_client.send(str(ir))
+    print(resp)