medmekk
/

rmsnorm_kernel

Model card Files Files and versions

rmsnorm_kernel / build.toml

EC2 Default User

updates

538355f about 1 month ago

history blame contribute delete

238 Bytes

	[general]
	name = "rmsnorm_kernel"
	universal = false

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

	[kernel.rmsnorm_kernel]
	backend = "cuda"
	src = [
	"rmsnorm_kernel/rmsnorm.cu",
	]
	depends = [ "torch"]