search

harvardnlp / genbmm

CUDA kernels for generalized matrix-multiplication in PyTorch
79 stars 13 forks source link

NameError: name '_genbmm' is not defined #13

Open HenryBao91 opened 2 years ago

HenryBao91 commented 2 years ago

I install the module by !pip3 install -qU git+https://github.com/harvardnlp/genbmm. And then, I run the example code :

import genbmm

a = torch.rand(10, 3, 4).cuda().requires_grad_(True)
b = torch.rand(10, 4, 5).cuda().requires_grad_(True)

# Log-Sum-Exp
c = genbmm.logbmm(a, b)

It occurs this error :

NameError: name '_genbmm' is not defined
srush commented 2 years ago

This means that the cuda extension packaged with the library failed to install. Was their an error message when you installed?