[general]
name = "kernels_paged_attention_metal"
universal = false

[torch]
src = [
  "torch-ext/torch_binding.cpp",
  "torch-ext/torch_binding.h"
]

[kernel.activation_metal]
backend = "metal"
src = [
  "paged-attention-metal/attention/paged_attention.metal",
  "paged-attention-metal/cache/copy_blocks.metal",
  "paged-attention-metal/cache/reshape_and_cache.metal",
  "paged-attention-metal/convert_fp8.metal",
  "paged-attention-metal/float8.metal",
  "paged-attention-metal/utils.metal",
  "paged-attention-metal/paged_attention.mm",
  "paged-attention-metal/cache.mm",
  "paged-attention-metal/convert_fp8.mm",
  "paged-attention-metal/device.mm",
]
include = [ "." ]
depends = [ "torch" ]