Function PYBIND11_MODULE

examples/cuda_cpp/rmsnorm/main.cpp:63–65

--- Pybind11 Module Definition --- Exposes the 'run' function to Python, making it callable as a C++ extension.

// --- Pybind11 Module Definition ---
// Exposes the 'run' function to Python, making it callable as a C++ extension.
PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
    m.def("run", &run, "RMSNorm kernel for hidden_size=4096 (BFloat16, CUDA, B200 Optimized)");
}