Function assign_col_time_ms

src/fused_mm_sampling/bench/speed_test.py:197–200

(df: pd.DataFrame)

def assign_col_time_ms(df: pd.DataFrame) -> pd.DataFrame:
    df["GPU Time (ms)"] = (df["GPU Time (sec)"] * 1e3).round(3)
    df["CPU Time (ms)"] = (df["CPU Time (sec)"] * 1e3).round(3)
    return df


def _as_torch_stream(cs: "nvbench.CudaStream") -> torch.cuda.ExternalStream: