blocksparse
blocksparse copied to clipboard
Error when importing blocksparse: `undefined symbol: _Z11GetCountSMsv`
I am having trouble importing your blocksparse library after compiling and pip-installing the built package. I am compiling using cuda 10 and the latest tf built from source, so I imagine this could be the issue as you mention using cuda 8 and tf 1.4.0. I'd like to try and get it working with the latest tf, as the rest of our pipelines and models are built upon the latest version
I found several references to GetCountSMs() and GetCountSMsVersion() in the source code, but could not find anything called _Z11GetCountSMsv
Thanks in advance for any help
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/home/ubuntu/blocksparse/blocksparse/__init__.py", line 3, in <module>
from blocksparse.utils import (
File "/home/ubuntu/blocksparse/blocksparse/utils.py", line 17, in <module>
_op_module = tf.load_op_library(os.path.join(data_files_path, 'blocksparse_ops.so'))
File "/home/ubuntu/.conda/envs/bs/lib/python3.7/site-packages/tensorflow_core/python/framework/load_library.py", line 61, in load_op_library
lib_handle = py_tf.TF_LoadLibrary(library_filename)
tensorflow.python.framework.errors_impl.NotFoundError: /home/ubuntu/blocksparse/blocksparse/blocksparse_ops.so: undefined symbol: _Z11GetCountSMsv