burtenshaw HF Staff commited on
Commit
0286af4
·
verified ·
1 Parent(s): 73f9ce9

Upload build.toml with huggingface_hub

Browse files
Files changed (1) hide show
  1. build.toml +20 -0
build.toml ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Qwen3-8B Custom Kernels - Build Configuration
2
+ [general]
3
+ name = "qwen3_kernels"
4
+ version = 1
5
+ backends = ["cuda"]
6
+
7
+ [general.cuda]
8
+
9
+ [torch]
10
+ src = [
11
+ "torch-ext/torch_binding.cpp",
12
+ "torch-ext/torch_binding.h",
13
+ ]
14
+
15
+ [kernel.rmsnorm]
16
+ backend = "cuda"
17
+ depends = ["torch"]
18
+ src = ["kernel_src/rmsnorm.cu"]
19
+ cuda-capabilities = ["9.0"]
20
+