search

FindDefinition / cumm

CUda Matrix Multiply library.
Apache License 2.0
70 stars 9 forks source link

ImportError: cannot import name 'TensorOpParams' from 'cumm.gemm.algospec.core' #3

Open SmBito opened 2 years ago

SmBito commented 2 years ago

Hi, I built cumm from source on Nvidia Jetson nano board, when I import cumm inside python, no errors appear. But when import TensorOpParams as following: from cumm.gemm.algospec.core import TensorOpParams I got the following error:

Traceback (most recent call last): File "", line 1, in File "/home/jetson/build_spconv/spconv3/spconv/spconv/init.py", line 15, in from . import build as _build File "/home/jetson/build_spconv/spconv3/spconv/spconv/build.py", line 24, in from spconv.core import SHUFFLE_SIMT_PARAMS, SHUFFLE_VOLTA_PARAMS, SHUFFLE_TURING_PARAMS File "/home/jetson/build_spconv/spconv3/spconv/spconv/core.py", line 18, in from cumm.gemm.algospec.core import TensorOpParams ImportError: cannot import name 'TensorOpParams' from 'cumm.gemm.algospec.core' (/home/jetson/build_spconv/spconv3/cumm/cumm/gemm/algospec/core.py)

Environment： NVIDIA Jetson Nano Developer Kit  (aarch64) Python: 3.7.5 cumm: 0.3.0 cuda: 10.2

Please help me!!! Thank you!

ChulanZhang commented 2 years ago

I got the same errors. Any solutions will be appreciated!

LexcaliburR commented 2 years ago

you can checkout to tags v0.2.8