-
-
Notifications
You must be signed in to change notification settings - Fork 823
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
Add cupy_backends/cuda/libs/cutensor.pxd
#3595
Conversation
97a16a5
to
1590d30
Compare
cupy.cutensor
cupy_backends/cuda/libs/cutensor.pxd
@kmaehashi PTAL! |
|
||
cdef class Handle: | ||
|
||
cdef void* _ptr |
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.
This PR changes the type of member variable _ptr
, because cutensorHandle_t
is declared in the pyx file.
4a1bc63
to
1590d30
Compare
Jenkins, test this please. |
Successfully created a job for commit 1590d30: |
Jenkins CI test (for commit 1590d30, target branch master) succeeded! |
cupy_setup_build.py
Outdated
@@ -83,6 +83,7 @@ | |||
'cupy.cuda.function', | |||
'cupy.cuda.stream', | |||
'cupy.cuda.texture', | |||
'cupy.cutensor', |
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.
Could you move this under cutensor module?
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.
Fixed.
LGTM except comment. |
LGTM! |
Successfully created a job for commit eb60ce8: |
Jenkins CI test (for commit eb60ce8, target branch master) succeeded! |
This PR adds
cupy_backends/cuda/libs/cutensor.pxd
and cythonizescupy.cutensor
.Further performance improvement will be worked in another PR: #3700