File size: 359 Bytes
df5d07c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | [general]
name = "universal"
license = "Apache-2.0"
backends = [
"rocm",
]
version = 1
edition = 5
[general.hub]
repo-id = "Ashiedu/universal"
[torch]
src = [
"torch-ext/torch_binding.cpp",
"torch-ext/torch_binding.h",
]
[kernel.universal_rocm]
backend = "rocm"
depends = ["torch"]
rocm-archs = [
"gfx1030",
]
src = ["universal_cuda/universal.cu"]
|