File size: 1,303 Bytes
29e93ec fdd8ac0 29e93ec |
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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
[general]
version = "0.0.1"
[torch]
name = "moe"
src = [
"core/scalar_type.hpp",
"ext-torch/registration.h",
"ext-torch/torch_binding.cpp",
"ext-torch/torch_binding.h"
]
include = [ "." ]
pyroot = "ext-torch"
pyext = [ "py", "json" ]
[kernel.moe]
capabilities = [ "7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0" ]
src = [
"cuda_compat.h",
"dispatch_utils.h",
"moe/moe_align_sum_kernels.cu",
"moe/topk_softmax_kernels.cu",
]
depends = [ "torch" ]
[kernel.moe-marlin]
capabilities = [ "8.0", "8.6", "8.7", "8.9", "9.0" ]
src = [
"core/exception.hpp",
"core/scalar_type.hpp",
"marlin-moe/marlin_moe_ops.cu",
"marlin-moe/marlin_kernels/marlin_moe_kernel_ku4.cu",
"marlin-moe/marlin_kernels/marlin_moe_kernel_ku8b128.cu",
"marlin-moe/marlin_kernels/marlin_moe_kernel.h",
"marlin-moe/marlin_kernels/marlin_moe_kernel_ku4.h",
"marlin-moe/marlin_kernels/marlin_moe_kernel_ku4b8.h",
"marlin-moe/marlin_kernels/marlin_moe_kernel_ku4b8.cu",
"marlin-moe/marlin_kernels/marlin_moe_kernel_ku8b128.h",
]
include = [ "." ]
depends = [ "torch" ]
[kernel.activation]
capabilities = [ "7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0" ]
src = [
"activation/activation_kernels.cu",
"activation/cuda_compat.h",
"activation/dispatch_utils.h",
]
depends = [ "torch" ]
|