build.toml

[general]
name = "bitnet_kernel"

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

[kernel.rmsnorm_kernel]
src = [
  "bitnet_kernel/bitnet_kernel.cu",
  "bitnet_kernel/bitnet_kernel.h
]
depends = [ "torch"]