Skip to content

Commit 6fc9dbf

Browse files
committed
Correcting errata in slides and removing unnecessary .types from pygraphblas code.
1 parent ca579c0 commit 6fc9dbf

File tree

2 files changed

+14
-14
lines changed

2 files changed

+14
-14
lines changed
Binary file not shown.

notebook-workspace/algorithms.py

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33
#==========================================================================
44
def neighborhood(graph, src, num_hops):
55
num_nodes = graph.nrows
6-
w = grb.Vector.sparse(grb.types.BOOL, num_nodes)
7-
v = grb.Vector.sparse(grb.types.BOOL, num_nodes)
6+
w = grb.Vector.sparse(grb.BOOL, num_nodes)
7+
v = grb.Vector.sparse(grb.BOOL, num_nodes)
88
w[src] = True
99
v.assign_scalar(True, mask=w)
1010

@@ -19,12 +19,12 @@ def neighborhood(graph, src, num_hops):
1919
def pagerank(A, damping = 0.85, tol = 1e-4, itermax = 100):
2020
n = A.nrows
2121

22-
r = grb.Vector.dense(grb.types.FP32, n, fill=1.0/n)
23-
t = grb.Vector.dense(grb.types.FP32, n)
24-
d = grb.Vector.dense(grb.types.FP32, n, fill=damping)
22+
r = grb.Vector.dense(grb.FP32, n, fill=1.0/n)
23+
t = grb.Vector.dense(grb.FP32, n)
24+
d = grb.Vector.dense(grb.FP32, n, fill=damping)
2525

26-
A.reduce_vector(out=d, accum=grb.types.FP32.DIV,
27-
mon=grb.types.FP32.PLUS_MONOID)
26+
A.reduce_vector(out=d, accum=grb.FP32.DIV,
27+
mon=grb.FP32.PLUS_MONOID)
2828

2929
teleport = (1 - damping) / n
3030

@@ -34,12 +34,12 @@ def pagerank(A, damping = 0.85, tol = 1e-4, itermax = 100):
3434
r[:] = teleport
3535
A.mxv(w,
3636
out=r,
37-
accum=grb.types.FP32.PLUS,
38-
semiring=grb.types.FP32.PLUS_SECOND,
37+
accum=grb.FP32.PLUS,
38+
semiring=grb.FP32.PLUS_SECOND,
3939
desc=grb.descriptor.T0)
4040
t -= r
41-
t.apply(grb.types.FP32.ABS, out=t)
42-
rdiff = t.reduce_float(grb.types.FP32.PLUS_MONOID)
41+
t.apply(grb.FP32.ABS, out=t)
42+
rdiff = t.reduce_float(grb.FP32.PLUS_MONOID)
4343
if rdiff <= tol:
4444
break
4545
return r
@@ -55,9 +55,9 @@ def triangle_count(A):
5555
A_nl = A_unw.offdiag()
5656

5757
# make sure of symmetry
58-
A_nl.eadd(A_nl, add_op=grb.types.UINT64.LOR, desc=grb.descriptor.T1, out=A_nl)
58+
A_nl.eadd(A_nl, add_op=grb.UINT64.LOR, desc=grb.descriptor.T1, out=A_nl)
5959

6060
# count triangles
61-
C = A_nl.mxm(A_nl, semiring=grb.types.UINT64.PLUS_TIMES, mask=A_nl)
62-
count = C.reduce_int(grb.types.UINT64.PLUS_MONOID)
61+
C = A_nl.mxm(A_nl, semiring=grb.UINT64.PLUS_TIMES, mask=A_nl)
62+
count = C.reduce_int(grb.UINT64.PLUS_MONOID)
6363
return (int)(count/6.0)

0 commit comments

Comments
 (0)