implement varargs (#83)

* implement varargs * clean up varargs
2026-07-28 11:25:04 +08:00 · 2016-06-04 16:22:10 -07:00
parent 2b52b91acb
commit f9aeb5d018
6 changed files with 74 additions and 24 deletions
@@ -83,6 +83,6 @@ def cond(x):
 def matrix_rank(M):
  return np.linalg.matrix_rank(M)

-@halo.remote([np.ndarray, None], [np.ndarray])
-def multi_dot(a):
+@halo.remote([np.ndarray], [np.ndarray])
+def multi_dot(*a):
  raise NotImplementedError