(fun, in1, in2)
| 200 | |
| 201 | |
| 202 | def cupy_adapter(fun, in1, in2): |
| 203 | tin1 = [cupy.from_dlpack(dltensor) for dltensor in in1] |
| 204 | tin2 = [cupy.from_dlpack(dltensor) for dltensor in in2] |
| 205 | tout1, tout2 = fun(tin1, tin2) |
| 206 | return [tout.__dlpack__() for tout in tout1], [tout.__dlpack__() for tout in tout2] |
| 207 | |
| 208 | |
| 209 | def cupy_wrapper(fun, synchronize): |
no test coverage detected