CMakeLists.txt
LICENSE
MANIFEST.in
README.md
pyproject.toml
requirements.txt
setup.py
vidrial/__init__.py
vidrial/cuda_utils/ABC_utils.cuh
vidrial/cuda_utils/allocator.cuh
vidrial/cuda_utils/copy.cuh
vidrial/cuda_utils/frg_copy.cuh
vidrial/cuda_utils/frgthr_tiling.cuh
vidrial/cuda_utils/gemm.cuh
vidrial/cuda_utils/launch.cuh
vidrial/cuda_utils/mask_cfg.cuh
vidrial/cuda_utils/perf_cfg.cuh
vidrial/cuda_utils/pipeline.cuh
vidrial/cuda_utils/softmax.cuh
vidrial/cuda_utils/swizzle_cfg.cuh
vidrial/cuda_utils/sympow.cuh
vidrial/cuda_utils/tprod.cuh
vidrial/cuda_utils/utilities.cuh
vidrial/cuda_utils/tests/copy_test.cu
vidrial/cuda_utils/tests/frgthr_tiling_test.cu
vidrial/cuda_utils/tests/ldsm_test.cu
vidrial/cuda_utils/tests/swizzle_test.cu
vidrial/cuda_utils/tests/tprod_test.cu
vidrial/cuda_utils/tests/utils_test.cu
vidrial/jit/__init__.py
vidrial/jit/binding.py
vidrial/jit/compiler.py
vidrial/jit/ctypes.py
vidrial/jit/jit.py
vidrial/jit/package.py
vidrial/jit/runtime.py
vidrial/jit/settings.py
vidrial/jit/timer.py
vidrial/jit/timing.py
vidrial/jit/timingcache.py
vidrial/jit/tuner.py
vidrial/jit/viz.py
vidrial/jit/static/__init__.py
vidrial/jit/static/algorithms.py
vidrial/jit/static/int.py
vidrial/jit/static/shape.py
vidrial/jit/static/stride.py
vidrial/jit/static/tuples.py
vidrial/jit/static/types.py
vidrial/jit/static/util.py
vidrial/jit/static/tests/test_layout.py
vidrial/jit/static/tests/test_shape.py
vidrial/jit/static/tests/test_tuple.py
vidrial/jit/tests/conftest.py
vidrial/jit/tests/test_compiler.py
vidrial/jit/tests/test_decorator.py
vidrial/jit/tests/test_jit.py
vidrial/jit/tests/test_timing_cache.py
vidrial/jit/tests/test_tuner.py
vidrial/kernels/__init__.py
vidrial/kernels/conftest.py
vidrial/kernels/mma_configurator.py
vidrial/kernels/add_one/__init__.py
vidrial/kernels/add_one/add_one.py
vidrial/kernels/add_one/add_one_kernels.cuh
vidrial/kernels/add_one/test.cu
vidrial/kernels/add_one/test.py
vidrial/kernels/copy/copy_cfg.cuh
vidrial/kernels/copy/copy_cfg_test.cu
vidrial/kernels/copy/copy_kernels.cuh
vidrial/kernels/flash/binding.py
vidrial/kernels/flash/compare.py
vidrial/kernels/flash/configurator.py
vidrial/kernels/flash/flash_cfg.cuh
vidrial/kernels/flash/interface.py
vidrial/kernels/flash/kernel.cuh
vidrial/kernels/flash/op.py
vidrial/kernels/flash/test.cu
vidrial/kernels/flash/test.py
vidrial/kernels/mma/__init__.py
vidrial/kernels/mma/binding.py
vidrial/kernels/mma/dimensions.py
vidrial/kernels/mma/kernel.cuh
vidrial/kernels/mma/mma_cfg.cuh
vidrial/kernels/mma/mma_cfg_test.cu
vidrial/kernels/mma/op.py
vidrial/kernels/mma/test.py
vidrial/kernels/mma_sympow_bwd/__init__.py
vidrial/kernels/mma_sympow_bwd/binding.py
vidrial/kernels/mma_sympow_bwd/dimensions.py
vidrial/kernels/mma_sympow_bwd/interface.py
vidrial/kernels/mma_sympow_bwd/kernel.cuh
vidrial/kernels/mma_sympow_bwd/op.py
vidrial/kernels/mma_sympow_bwd/sympow_C_mma_cfg.cuh
vidrial/kernels/mma_sympow_bwd/test.py
vidrial/kernels/reduce/__init__.py
vidrial/kernels/reduce/binding.py
vidrial/kernels/reduce/kernel.cuh
vidrial/kernels/reduce/reduce_utils.cuh
vidrial/kernels/reduce/test.cu
vidrial/kernels/reduce/test.py
vidrial/kernels/sympow/__init__.py
vidrial/kernels/sympow/binding.py
vidrial/kernels/sympow/dimensions.py
vidrial/kernels/sympow/interface.py
vidrial/kernels/sympow/kernel.cuh
vidrial/kernels/sympow/op.py
vidrial/kernels/sympow/sympow_cfg.cuh
vidrial/kernels/sympow/sympow_cfg_test.cu
vidrial/kernels/sympow/test.cu
vidrial/kernels/sympow/test.py
vidrial/kernels/sympow_bwd/__init__.py
vidrial/kernels/sympow_bwd/binding.py
vidrial/kernels/sympow_bwd/kernel.cuh
vidrial/kernels/sympow_bwd/op.py
vidrial/kernels/sympow_bwd/sympow_bwd.cuh
vidrial/kernels/sympow_bwd/test.cu
vidrial/kernels/sympow_bwd/test.py
vidrial/kernels/sympow_bwd/tprod_bwd.cuh
vidrial/kernels/sympow_mma/__init__.py
vidrial/kernels/sympow_mma/binding.py
vidrial/kernels/sympow_mma/dimensions.py
vidrial/kernels/sympow_mma/interface.py
vidrial/kernels/sympow_mma/kernel.cuh
vidrial/kernels/sympow_mma/op.py
vidrial/kernels/sympow_mma/sympow_mma_cfg.cuh
vidrial/kernels/sympow_mma/test.cu
vidrial/kernels/sympow_mma/test.py
vidrial/kernels/sympow_mma/test_for_dist.py
vidrial/py_utils/__init__.py
vidrial/py_utils/common.py
vidrial/py_utils/demo.py
vidrial/py_utils/gpu.py
vidrial/py_utils/test_utils.py