Function main

src/fused_mm_sampling/modal_lib/modal_ncu.py:91–99

()

        n_hidden_states=str(cfg.n_hidden_states),
        case=cfg.case,
        n_procs=cfg.n_procs,
        mode=cfg.ncu_mode,
        gpu_name=cfg.gpu,
    )