-
Notifications
You must be signed in to change notification settings - Fork 70
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.
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
Single processor implementation for linalg.solve
#568
Single processor implementation for linalg.solve
#568
Conversation
for more information, see https://pre-commit.ci
…unumeric into almogsegal-add-cunumeric-solve
@rohany I added an optional output parameter to the internal API as I don't want to deviate from the NumPy API: https://github.com/nv-legate/cunumeric/pull/568/files#diff-809207875808b833ab90102d3ef7268ccff656c308004f73a82822903bf2b434R635 note that this function bypasses all argument checks in the user-facing version, so use it at your own risk. |
LGTM |
#endif | ||
|
||
size_t a_strides[2]; | ||
VAL* a = a_array.read_write_accessor<VAL, 2>(a_shape).ptr(a_shape, a_strides); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can we add a debugging check here, that
a_strides[0] == m && a_strides[1] == 1
and similarly for b_strides
below?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Note that the assertion should be a_strides[0] == 1 && a_strides[1] == m
, as this task assumes the instances to have the Fortran order.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
One more debugging check suggested, otherwise looks good
This extends #518 with some extra changes to make the code fully functional.