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

matmul fails with ints on GPU #657

Closed
coquelin77 opened this issue Aug 25, 2020 · 0 comments · Fixed by #658
Closed

matmul fails with ints on GPU #657

coquelin77 opened this issue Aug 25, 2020 · 0 comments · Fixed by #658

Comments

@coquelin77
Copy link
Member

To Reproduce

import heat as ht

a = ht.random.randint(0, 30, size=(30, 30), device="gpu", split=0)
b = ht.random.randint(0, 30, size=(30, 30), device="gpu", split=0)
ht.dot(a, b)

result:

RuntimeError: "addmm_cuda" not implemented for 'Int'

Traceback (most recent call last):
  File "dot-test.py", line 5, in <module>
    ht.dot(a, b)
  File "/p/home/jusers/coquelin1/hdfml/heat/heat/heat/core/linalg/basics.py", line 56, in dot
    ret = matmul(a, b)
  File "/p/home/jusers/coquelin1/hdfml/heat/heat/heat/core/linalg/basics.py", line 501, in matmul
    c=c._DNDarray__array,
RuntimeError: The following operation failed in the TorchScript interpreter.
Traceback of TorchScript (most recent call last):
  File "/p/home/jusers/coquelin1/hdfml/heat/heat/heat/core/linalg/basics.py", line 803, in __mm_c_block_setter
                    c_start0 = a_start0
                    c_start1 = b_start1
                    c[c_start0 : c_start0 + mB, c_start1 : c_start1 + nB] += a_block @ b_block
                                                                             ~~~~~~~~~~~~~~~~~ <--- HERE
RuntimeError: "addmm_cuda" not implemented for 'Int'

Expected behavior
Should run without failure

Version Info
current master

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

Successfully merging a pull request may close this issue.

1 participant