File size: 332 Bytes
0286af4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | # Qwen3-8B Custom Kernels - Build Configuration
[general]
name = "qwen3_kernels"
version = 1
backends = ["cuda"]
[general.cuda]
[torch]
src = [
"torch-ext/torch_binding.cpp",
"torch-ext/torch_binding.h",
]
[kernel.rmsnorm]
backend = "cuda"
depends = ["torch"]
src = ["kernel_src/rmsnorm.cu"]
cuda-capabilities = ["9.0"]
|