Update for new Python source specification
Browse files- build.toml +1 -3
build.toml
CHANGED
@@ -8,9 +8,7 @@ src = [
|
|
8 |
"ext-torch/torch_binding.cpp",
|
9 |
"ext-torch/torch_binding.h"
|
10 |
]
|
11 |
-
|
12 |
-
"ext-torch/__init__.py"
|
13 |
-
]
|
14 |
|
15 |
[kernel.activation]
|
16 |
capabilities = [ "7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0" ]
|
|
|
8 |
"ext-torch/torch_binding.cpp",
|
9 |
"ext-torch/torch_binding.h"
|
10 |
]
|
11 |
+
pyroot = "ext-torch"
|
|
|
|
|
12 |
|
13 |
[kernel.activation]
|
14 |
capabilities = [ "7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0" ]
|