Skip to content
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

Merged
merged 2 commits into from
Aug 6, 2020

Conversation

asi1024
Copy link
Member

@asi1024 asi1024 commented Jul 12, 2020

This PR adds cupy_backends/cuda/libs/cutensor.pxd and cythonizes cupy.cutensor.
Further performance improvement will be worked in another PR: #3700

@kmaehashi kmaehashi self-assigned this Jul 13, 2020
@kmaehashi kmaehashi added this to the v8.0.0rc1 milestone Jul 15, 2020
@kmaehashi kmaehashi added the blocking Issue/pull-request is mandatory for the upcoming release label Jul 20, 2020
@kmaehashi kmaehashi removed the blocking Issue/pull-request is mandatory for the upcoming release label Jul 28, 2020
@asi1024 asi1024 removed this from the v8.0.0b5 milestone Jul 29, 2020
@asi1024 asi1024 mentioned this pull request Jul 31, 2020
@asi1024 asi1024 added the cat:code-fix Code refactoring that do not change behavior label Jul 31, 2020
@asi1024 asi1024 changed the title [WIP] Cythonize cupy.cutensor Cythonize cupy.cutensor Jul 31, 2020
@asi1024 asi1024 changed the title Cythonize cupy.cutensor Add cupy_backends/cuda/libs/cutensor.pxd Jul 31, 2020
@asi1024
Copy link
Member Author

asi1024 commented Jul 31, 2020

@kmaehashi PTAL!


cdef class Handle:

cdef void* _ptr
Copy link
Member Author

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.

@asi1024
Copy link
Member Author

asi1024 commented Aug 1, 2020

Jenkins, test this please.

@pfn-ci-bot
Copy link
Collaborator

Successfully created a job for commit 1590d30:

@chainer-ci
Copy link
Member

Jenkins CI test (for commit 1590d30, target branch master) succeeded!

@@ -83,6 +83,7 @@
'cupy.cuda.function',
'cupy.cuda.stream',
'cupy.cuda.texture',
'cupy.cutensor',
Copy link
Member

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?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Fixed.

@kmaehashi
Copy link
Member

LGTM except comment.

@kmaehashi
Copy link
Member

LGTM!
pfnCI, test this please.

@pfn-ci-bot
Copy link
Collaborator

Successfully created a job for commit eb60ce8:

@kmaehashi kmaehashi added st:test-and-merge (deprecated) Ready to merge after test pass. cat:code-fix Code refactoring that do not change behavior and removed cat:code-fix Code refactoring that do not change behavior labels Aug 6, 2020
@chainer-ci
Copy link
Member

Jenkins CI test (for commit eb60ce8, target branch master) succeeded!

@mergify mergify bot merged commit 594ecc6 into cupy:master Aug 6, 2020
@asi1024 asi1024 deleted the cythonize-cutensor branch August 6, 2020 10:24
@asi1024 asi1024 added this to the v8.0.0rc1 milestone Aug 6, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cat:code-fix Code refactoring that do not change behavior st:test-and-merge (deprecated) Ready to merge after test pass.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants