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/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.c
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/fragment.py
spio/generators/gen_specs.py
spio/generators/generators.py
spio/generators/index.py
spio/generators/macros.py
spio/generators/params.py
spio/generators/tensor.py
spio/include/__init__.py
spio/include/spio/__init__.py
spio/include/spio/fragment.cuh
spio/include/spio/index.h
spio/include/spio/ldmatrix.cuh
spio/include/spio/mma.cuh
spio/include/spio/pipeline.h
spio/include/spio/tensor.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_tests/Conv2dGw8Params.dat
spio/src_tests/__init__.py
spio/src_tests/add.cu
spio/src_tests/ldmatrix.cu
spio/src_tests/memcpy_simple.cu
spio/src_tests/mma.cu
spio/src_tests/preprocess_data_file.py
spio/src_tests/row_memcpy.cu
spio/src_tests/run_test.py
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/parse_dataclass.py
spio/util/parse_kwargs.py
tests/test_conv2d_gw8.py
tests/test_index.py
tests/test_kernel.py
tests/test_ldmatrix.py
tests/test_nvcc.py
tests/test_stats.py