#! /usr/bin/env nu

def main [
    --mmq_data (-m): path = mmq.csv
    --blas_data (-b): path = blas.csv
    --tuned_data (-t): path = tuned.csv
]: nothing -> string {
    open $mmq_data
    | select model_filename model_n_params n_ubatch n_prompt avg_ts
    | rename -c {avg_ts: mmq_ts}
    | merge (
        open $blas_data
        | select avg_ts
        | rename -c {avg_ts: blas_ts}
    )
    | merge (
        open $tuned_data
        | select avg_ts
        | rename -c {avg_ts: tuned_ts}
    )
    | insert mmq/blas {|r| ($r.mmq_ts / $r.blas_ts - 1) * 100 }
    | insert mmq% {|r| ($r.mmq_ts / ([$r.mmq_ts $r.blas_ts] | math max)) * 100 }
    | insert blas% {|r| ($r.blas_ts / ([$r.mmq_ts $r.blas_ts] | math max)) * 100 }
    | insert tuned% {|r| ($r.tuned_ts / ([$r.mmq_ts $r.blas_ts] | math max)) * 100 }
    | to csv
}