replicate
/

mrope_get_position_ids

Model card Files Files and versions

mrope_get_position_ids / build.toml

danieldk's picture

danieldk HF Staff

Sync with latest kernel-builder changes

b865e0f over 1 year ago

Raw History Blame Contribute Delete

345 Bytes

	[general]
	name = "get_position_ids"

	[torch]
	src = [
	"ext-torch/torch_binding.cpp",
	"ext-torch/torch_binding.h",
	]
	include = ["."]
	pyext = ["py", "json"]

	[kernel.get_position_ids]
	cuda-capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0"]
	src = ["get_position_ids/get_position_ids.cu"]
	include = ["."]
	depends = ["torch"]