kernels-community/megablocks

1[general]2name = "megablocks"3universal = false4 5[torch]6src = [7  "torch-ext/torch_binding.cpp",8  "torch-ext/torch_binding.h"9]10 11[kernel.megablocks]12backend = "cuda"13cuda-capabilities = [14    "7.0",15    "7.2",16    "7.5",17    "8.0",18    "8.6",19    "8.7",20    "8.9",21    "9.0",22    "10.0",23    "10.1",24    "11.8",25    "12.0"26]27depends = ["torch", "cutlass_3_8"]28src = [29    "csrc/new_cumsum.h",30    "csrc/new_cumsum.cu",31    "csrc/new_histogram.h",32    "csrc/new_histogram.cu",33    "csrc/new_indices.h",34    "csrc/new_indices.cu",35    "csrc/new_replicate.cu",36    "csrc/new_replicate.h",37    "csrc/new_sort.h",38    "csrc/new_sort.cu",39    # vendored grouped gemm40    "csrc/grouped_gemm/fill_arguments.cuh",41    "csrc/grouped_gemm/grouped_gemm.cu",42    "csrc/grouped_gemm/grouped_gemm.h",43]