LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
perf/__init__.py
perf/_checks.py
perf/_inspect.py
perf/_precision.py
perf/_timing.py
perf/_utils.py
perf/benchmark.py
perf/compare_w_triton.py
perf/db.py
perf/benchmarks/__init__.py
perf/benchmarks/fla.py
perf/benchmarks/flash.py
perf/benchmarks/runs.py
perf/benchmarks/sdpa.py
retention/__init__.py
retention/_config.py
retention/_utils.py
retention/create_inputs.py
retention/cuda.py
retention/kernelgen.py
retention/power_retention_maker.py
retention/reference.py
retention/triton.py
retention/vidrial.py
retention/vidrial_fused_reference.py
retention/vidrial_reference.py
retention.egg-info/PKG-INFO
retention.egg-info/SOURCES.txt
retention.egg-info/dependency_links.txt
retention.egg-info/requires.txt
retention.egg-info/top_level.txt
retention/_attention/__init__.py
retention/_attention/create_inputs.py
retention/_attention/cuda.py
retention/_attention/reference.py
retention/_attention/triton.py
retention/_discumsum/__init__.py
retention/_discumsum/create_inputs.py
retention/_discumsum/cuda.py
retention/_discumsum/reference.py
retention/_discumsum/triton.py
retention/_query_state/__init__.py
retention/_query_state/create_inputs.py
retention/_query_state/cuda.py
retention/_query_state/reference.py
retention/_query_state/reference_vidrial.py
retention/_query_state/reference_vidrial_fused.py
retention/_query_state/triton.py
retention/_query_state/vidrial.py
retention/_query_state/vidrial_fused.py
retention/_query_state/vidrial_fused_kernel.py
retention/_update_state/__init__.py
retention/_update_state/create_inputs.py
retention/_update_state/cuda.py
retention/_update_state/reference.py
retention/_update_state/reference_vidrial.py
retention/_update_state/reference_vidrial_fused.py
retention/_update_state/triton.py
retention/_update_state/vidrial.py
retention/_update_state/vidrial_fused.py
retention/experimental/__init__.py