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

gpuarray fail to transpose and flatten after cublas.Sgemm #325

Open
SuperbTUM opened this issue Oct 7, 2021 · 0 comments
Open

gpuarray fail to transpose and flatten after cublas.Sgemm #325

SuperbTUM opened this issue Oct 7, 2021 · 0 comments

Comments

@SuperbTUM
Copy link

SuperbTUM commented Oct 7, 2021

Problem

I am using Colab to do GPU programming. I came up with a problem when I am using matrix multiplication function Sgemm. The three inputs of the function would be gpuarrays. Once I got the result, I tried to transform the result gpuarray (which is a flatten one) by first transposing (.T) and second flattening (ravel()). I think this could get a matrix that is completely different from the raw output since I rearrange the order. But the truth is that I still got the identical matrix. Therefore, I want to know how to rearrange the order of gpuarray. Thanks in advance.

Environment

I think the environment is Linux 5.4.104+ and Python version is 3.7.12. CUDA is integrated in Colab with version Cuda 11.1. Pycuda version is 2021.1, and scikit-cuda version is 0.5.3.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant