Skip to content

A
alpha-mind
Commit cbc97303 authored by Dr.李's avatar Dr.李
Browse files
Options

added nb.njit for sqr sum

parent 1bf94321
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
alphamind/utilities.py
View file @ cbc97303
...@@ -95,6 +95,7 @@ def simple_abssum(x, axis=0): ...@@ -95,6 +95,7 @@ def simple_abssum(x, axis=0):
return res return res
@nb.njit(nogil=True, cache=True)
def simple_sqrsum(x, axis=0): def simple_sqrsum(x, axis=0):
length, width = x.shape length, width = x.shape
...@@ -175,6 +176,7 @@ def agg_sum(groups, x): ...@@ -175,6 +176,7 @@ def agg_sum(groups, x):
return res return res
@nb.njit(nogil=True, cache=True)
def agg_sqrsum(groups, x): def agg_sqrsum(groups, x):
max_g = groups.max() max_g = groups.max()
length, width = x.shape length, width = x.shape
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment