[general]
name = "rmsnorm_kernel"

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

[kernel.activation]
src = [
  "rmsnorm_kernel/rmsnorm.cpp",
]

depends = [ "torch"]

# If the kernel is only supported on specific capabilities, set the
# cuda-capabilities option:
#
# cuda-capabilities = [ "9.0", "10.0", "12.0" ]