README.md
setup.py
sptr/__init__.py
sptr/functional.py
sptr/modules.py
sptr/position_embedding.py
sptr/utils.py
sptr.egg-info/PKG-INFO
sptr.egg-info/SOURCES.txt
sptr.egg-info/dependency_links.txt
sptr.egg-info/requires.txt
sptr.egg-info/top_level.txt
sptr/src/pointops_api.cpp
sptr/src/attention/attention_cuda.cpp
sptr/src/attention/attention_cuda_kernel.cu
sptr/src/precompute/precompute.cpp
sptr/src/precompute/precompute_cuda_kernel.cu
sptr/src/rpe/relative_pos_encoding_cuda.cpp
sptr/src/rpe/relative_pos_encoding_cuda_kernel.cu
test/test.py
test/test_attention_op_step1.py
test/test_attention_op_step2.py
test/test_precompute_all.py
test/test_relative_pos_encoding_op_step1.py
test/test_relative_pos_encoding_op_step1_all.py
test/test_relative_pos_encoding_op_step2.py