LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
spio/__init__.py
spio.egg-info/PKG-INFO
spio.egg-info/SOURCES.txt
spio.egg-info/dependency_links.txt
spio.egg-info/requires.txt
spio.egg-info/top_level.txt
spio/compiler/__init__.py
spio/compiler/arch.py
spio/compiler/compile.py
spio/compiler/compile_kernel.py
spio/compiler/compile_nvcc.py
spio/compiler/compiler_pool.py
spio/cuda/__init__.py
spio/cuda/cdriver.pxd
spio/cuda/driver.pyi
spio/cuda/driver.pyx
spio/cuda/nvrtc_ctypes.py
spio/functional/__init__.py
spio/functional/conv2d_gw8_function.py
spio/generators/__init__.py
spio/generators/async_strip_loader.py
spio/generators/checkerboard.py
spio/generators/compound_index.py
spio/generators/data_type.py
spio/generators/dim.py
spio/generators/dims.py
spio/generators/fold.py
spio/generators/fragment.py
spio/generators/fragment_index.py
spio/generators/fragment_type.py
spio/generators/gen_specs.py
spio/generators/generators.py
spio/generators/macros.py
spio/generators/matmul.py
spio/generators/params.py
spio/generators/subindex_protocol.py
spio/generators/tensor.py
spio/include/__init__.py
spio/include/spio.cuh
spio/include/spio/__init__.py
spio/include/spio/allocator.h
spio/include/spio/async_strip_loader.cuh
spio/include/spio/checkerboard_index.h
spio/include/spio/compound_index.h
spio/include/spio/compound_index_base.h
spio/include/spio/coordinates.h
spio/include/spio/dim.h
spio/include/spio/dim_info.h
spio/include/spio/fifo.cuh
spio/include/spio/fragment.cuh
spio/include/spio/fragment_index.h
spio/include/spio/fragment_load_index.h
spio/include/spio/fragment_mma.cuh
spio/include/spio/ldmatrix.cuh
spio/include/spio/macros.h
spio/include/spio/mathutil.h
spio/include/spio/memory.cuh
spio/include/spio/meta.h
spio/include/spio/mma.cuh
spio/include/spio/pipeline.h
spio/include/spio/semaphore.cuh
spio/include/spio/strip_loader_params.h
spio/include/spio/tensor.h
spio/include/spio/typed_dims.h
spio/kernels/__init__.py
spio/kernels/conv2d_gw8_kernel.py
spio/kernels/conv2d_gw8_params.py
spio/kernels/conv2d_gw8_wgrad_kernel.py
spio/kernels/conv2d_stats.py
spio/kernels/kernel.py
spio/kernels/kernel_cache.py
spio/kernels/kernel_factory.py
spio/kernels/kernel_key.py
spio/kernels/kernel_params_logger.py
spio/kernels/kernel_util.py
spio/kernels/launch_params.py
spio/kernels/params.py
spio/kernels/performance_model_cache.py
spio/kernels/stats.py
spio/layers/__init__.py
spio/layers/conv2d_gw8.py
spio/layers/make.py
spio/reflection/__init__.py
spio/reflection/arg_info.py
spio/reflection/conv2d_gw8_reflection.py
spio/reflection/reflection.py
spio/src/__init__.py
spio/src/conv2d_gw8.cu
spio/src/conv2d_gw8_wgrad.cu
spio/src/layernorm_2d.cu
spio/src_tests/Conv2dGw8Params.dat
spio/src_tests/__init__.py
spio/src_tests/add.cu
spio/src_tests/dim_test_common.h
spio/src_tests/dim_test_types.h
spio/src_tests/fifo.cu
spio/src_tests/index.cu
spio/src_tests/ldmatrix.cu
spio/src_tests/memcpy_simple.cu
spio/src_tests/mma.cu
spio/src_tests/mma_checkerboard_16c.cu
spio/src_tests/preprocess_data_file.py
spio/src_tests/row_memcpy.cu
spio/src_tests/run_test.py
spio/src_tests/semaphore.cu
spio/src_tests/tutorial.h
spio/src_tests/utest.h
spio/transform/__init__.py
spio/transform/_transform.py
spio/util/__init__.py
spio/util/cache_dir.py
spio/util/class_names.py
spio/util/close.py
spio/util/device_info.py
spio/util/interval_timer.py
spio/util/load_parameter_set.py
spio/util/logger.py
spio/util/math.py
spio/util/memory_format.py
spio/util/parse_dataclass.py
spio/util/parse_kwargs.py
spio/util/tensor_format.py
spio/util/test_matrices.py
tests/test_conv2d_gw8.py
tests/test_cpp.py
tests/test_cuda_driver.py
tests/test_dtype.py
tests/test_kernel.py
tests/test_ldmatrix.py
tests/test_nvcc.py
tests/test_stats.py
tests/test_synchronization.py
tests/test_tutorial.py
tests/test_util.py