| 192 | |
| 193 | |
| 194 | class Request(object): |
| 195 | def __init__(self): |
| 196 | self.req = None |
| 197 | self.tensor = None |
| 198 | self.WaitFunction = All2All_Scatter_Wait |
| 199 | |
| 200 | def wait(self): |
| 201 | ret = self.WaitFunction.apply(*self.tensor) |
| 202 | self.req = None |
| 203 | self.tensor = None |
| 204 | return ret |
| 205 | |
| 206 | |
| 207 | class All2All_ScatterList_Req(Function): |