Function PYBIND11_MODULE

examples/cutlass/gemm/main.cpp:19–21

}

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
    m.def("run", &run, "CUTLASS GEMM");
}