wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 18:45:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

[api] Second round of 1.0 API changes: exceptions, num_return_vals (#10377)

Browse Source
This commit is contained in:
Eric Liang
2020-08-28 19:57:02 -07:00
committed by GitHub
parent b1f3c9e10e
commit 2a204260a8
37 changed files with 180 additions and 204 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/experimental/array/distributed/linalg.py
+5 -5
View File
@@ -7,7 +7,7 @@ from . import core
__all__ = ["tsqr", "modified_lu", "tsqr_hr", "qr"]
@ray.remote(num_return_vals=2)
@ray.remote(num_returns=2)
def tsqr(a):
"""Perform a QR decomposition of a tall-skinny matrix.
@@ -83,7 +83,7 @@ def tsqr(a):
# TODO(rkn): This is unoptimized, we really want a block version of this.
# This is Algorithm 5 from
# http://www.eecs.berkeley.edu/Pubs/TechRpts/2013/EECS-2013-175.pdf.
@ray.remote(num_return_vals=3)
@ray.remote(num_returns=3)
def modified_lu(q):
"""Perform a modified LU decomposition of a matrix.
@@ -121,7 +121,7 @@ def modified_lu(q):
return ray.get(core.numpy_to_dist.remote(ray.put(L))), U, S
@ray.remote(num_return_vals=2)
@ray.remote(num_returns=2)
def tsqr_hr_helper1(u, s, y_top_block, b):
y_top = y_top_block[:b, :b]
s_full = np.diag(s)
@@ -137,7 +137,7 @@ def tsqr_hr_helper2(s, r_temp):
# This is Algorithm 6 from
# http://www.eecs.berkeley.edu/Pubs/TechRpts/2013/EECS-2013-175.pdf.
@ray.remote(num_return_vals=4)
@ray.remote(num_returns=4)
def tsqr_hr(a):
q, r_temp = tsqr.remote(a)
y, u, s = modified_lu.remote(q)
@@ -160,7 +160,7 @@ def qr_helper2(y_ri, a_rc):
# This is Algorithm 7 from
# http://www.eecs.berkeley.edu/Pubs/TechRpts/2013/EECS-2013-175.pdf.
@ray.remote(num_return_vals=2)
@ray.remote(num_returns=2)
def qr(a):
m, n = a.shape[0], a.shape[1]
python/ray/experimental/array/remote/linalg.py
+7 -7
View File
@@ -18,12 +18,12 @@ def solve(a, b):
return np.linalg.solve(a, b)
@ray.remote(num_return_vals=2)
@ray.remote(num_returns=2)
def tensorsolve(a):
raise NotImplementedError
@ray.remote(num_return_vals=2)
@ray.remote(num_returns=2)
def tensorinv(a):
raise NotImplementedError
@@ -63,22 +63,22 @@ def det(a):
return np.linalg.det(a)
@ray.remote(num_return_vals=3)
@ray.remote(num_returns=3)
def svd(a):
return np.linalg.svd(a)
@ray.remote(num_return_vals=2)
@ray.remote(num_returns=2)
def eig(a):
return np.linalg.eig(a)
@ray.remote(num_return_vals=2)
@ray.remote(num_returns=2)
def eigh(a):
return np.linalg.eigh(a)
@ray.remote(num_return_vals=4)
@ray.remote(num_returns=4)
def lstsq(a, b):
return np.linalg.lstsq(a)
@@ -88,7 +88,7 @@ def norm(x):
return np.linalg.norm(x)
@ray.remote(num_return_vals=2)
@ray.remote(num_returns=2)
def qr(a):
return np.linalg.qr(a)