Skip to content

Add support for arrayanalysis of tuple args. #7398

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
sklam merged 16 commits into from
Oct 6, 2021
Merged

Add support for arrayanalysis of tuple args. #7398

Changes from 1 commit
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
36a9da7
Add support for arrayanalysis of tuple args.
DrTodd13 Sep 13, 2021
85e5548
Support tuples of integer literals in alias analysis.
DrTodd13 Sep 17, 2021
c3406ce
Support tuples that are combinations of integer vars and literals.
DrTodd13 Sep 17, 2021
435d434
Add test for pndindex of tuple arg with mix of literal and non-litera…
DrTodd13 Sep 17, 2021
6be6198
Fix flake8 problems.
DrTodd13 Sep 17, 2021
96981f2
Force index vars for pndindex to be constructed as tuples for 1 dimen…
DrTodd13 Sep 17, 2021
1f922eb
Add tests for 1d pndindex.
DrTodd13 Sep 17, 2021
98a6923
Add comment for force_tuple param of make_index_var.
DrTodd13 Oct 5, 2021
68047d3
Use numba.literally inside numba function to pass to another to trigg…
DrTodd13 Oct 5, 2021
cdd7a26
Add test and fix mistake in new parfor tests.
DrTodd13 Oct 5, 2021
8c06789
fix scope issue
DrTodd13 Oct 5, 2021
29c1440
Remove unneeded literally.
DrTodd13 Oct 5, 2021
ed34eb5
more arg tuple literal fixes
DrTodd13 Oct 6, 2021
99dec34
Add manual test of mixed tuple of int and literal.
DrTodd13 Oct 6, 2021
391893a
Add integer argument test for pndindex on 2d array.
DrTodd13 Oct 6, 2021
e39f6c9
Add ability for pndindex to take an integer argument in which case it…
DrTodd13 Oct 6, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
more arg tuple literal fixes
  • Loading branch information
@DrTodd13
DrTodd13 committed Oct 6, 2021
commit ed34eb56c6cf1aa6d8873eaf6c17852cf62fbee8
38 changes: 11 additions & 27 deletions numba/tests/test_parfors.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1864,7 +1864,7 @@ def test_impl(x, sz):
x[i] = 1
return x
sz = (10, 5)
self.check(test_impl, np.empty(sz), (10,3))
self.check(test_impl, np.zeros(sz), (10, 3))

def test_tuple_for_pndindex(self):
def test_impl(x):
Expand All @@ -1873,50 +1873,34 @@ def test_impl(x):
x[i] = 1
return x
sz = (10, 5)
self.check(test_impl, np.empty(sz))

def test_tuple_for_pndindex_with_literally(self):
def test_impl(x):
sz = (10, numba.literally(5))
for i in numba.pndindex(sz):
x[i] = 1
return x
sz = (10,5)
self.check(test_impl, np.empty(sz))
self.check(test_impl, np.zeros(sz))

def test_tuple_arg_literal(self):
@njit(parallel=True)
def test_impl(x, sz):
def test_impl(x, first):
sz = (first, 5)
for i in numba.pndindex(sz):
x[i] = 1
return x

def test_driver(impl_arg, first):
sz = (first, 5)
return impl_arg(np.empty(sz), sz)

self.check(test_driver, test_impl, 10)
sz = (10, 5)
self.check(test_impl, np.zeros(sz), 10)

def test_tuple_arg_1d(self):
def test_impl(x, sz):
for i in numba.pndindex(sz):
x[i] = 1
return x
sz = (10,)
self.check(test_impl, np.empty(sz), sz)
self.check(test_impl, np.zeros(sz), sz)

def test_tuple_arg_1d_literal(self):
@njit(parallel=True)
def test_impl(x, sz):
def test_impl(x):
sz = (10,)
for i in numba.pndindex(sz):
x[i] = 1
return x
sz = (10,)
self.check(test_impl, np.zeros(sz))

Comment thread
stuartarchibald marked this conversation as resolved.
Resolved
def test_driver(impl_arg):
sz = (10,)
return impl_arg(np.empty(sz), sz)

self.check(test_driver, test_impl)

@skip_parfors_unsupported
class TestParforsLeaks(MemoryLeakMixin, TestParforsBase):
Expand Down