README.md
pyproject.toml
coda/__init__.py
coda/core/__init__.py
coda/core/elementwise/__init__.py
coda/core/elementwise/cross_entropy.py
coda/core/elementwise/functional.py
coda/core/elementwise/rope.py
coda/core/elementwise/templates.py
coda/core/epilogue/__init__.py
coda/core/epilogue/activation.py
coda/core/epilogue/affine.py
coda/core/epilogue/base.py
coda/core/epilogue/epi_ops.py
coda/core/epilogue/lse.py
coda/core/epilogue/qknorm.py
coda/core/epilogue/utils.py
coda/core/gemm/__init__.py
coda/core/gemm/functional.py
coda/core/gemm/gemm_interface.py
coda/core/gemm/registry.py
coda/core/ops/__init__.py
coda/core/ops/benchmark_utils.py
coda/core/ops/creation_utils.py
coda/core/ops/debug_utils.py
coda/core/ops/dtype_utils.py
coda/core/ops/epilogue_utils.py
coda/core/ops/gemm_utils.py
coda/core/ops/launch_utils.py
coda/core/ops/layout_utils.py
coda/core/ops/math_utils.py
coda/core/ops/memory_utils.py
coda/core/ops/misc_utils.py
coda/core/ops/pipeline_utils.py
coda/core/ops/profiling_utils.py
coda/core/ops/reduction_utils.py
coda/core/ops/struct_utils.py
coda/core/ops/torch_utils.py
coda/kernels/__init__.py
coda/kernels/benchmarks/__init__.py
coda/kernels/benchmarks/bench_utils.py
coda/kernels/benchmarks/block.py
coda/kernels/benchmarks/gemm.py
coda/kernels/benchmarks/quack_utils.py
coda/kernels/benchmarks/trace_utils.py
coda/kernels/benchmarks/trainstation_utils.py
coda/kernels/benchmarks/torchtitan_extension/torchtitan_train.py
coda/kernels/functional/__init__.py
coda/kernels/functional/cross_entropy.py
coda/kernels/functional/qknorm_rope.py
coda/kernels/functional/swiglu.py
coda/kernels/gens/__init__.py
coda/kernels/gens/gpt.py
coda/kernels/refs/__init__.py
coda/kernels/refs/gpt.py
coda/kernels/refs/gpt2.py
coda/kernels/tests/__init__.py
coda/kernels/tests/gpt.py
coda/models/__init__.py
coda/models/gpt.py
coda/models/gpt_ref.py
coda/models/ops.py
coda/models/ops2.py
coda/models/tests.py
coda_kernels.egg-info/PKG-INFO
coda_kernels.egg-info/SOURCES.txt
coda_kernels.egg-info/dependency_links.txt
coda_kernels.egg-info/requires.txt
coda_kernels.egg-info/top_level.txt