diff --git a/README.md b/README.md index 1be0575d..3fca79fc 100644 --- a/README.md +++ b/README.md @@ -169,7 +169,7 @@ Program usage: --m=[:max height[:step]] Height of GEMM problem (number of rows of C). May specify a range with optional step size. --n=[:max width[:step]] Width of GEMM problem (number of columns of C). May specify a range with optional step size. --k=[:max depth[:step]] Size of inner dimension of A and B. May specify a range with optional step size. - --kernels=<{s|d|h|i|wmma}_gemm_{nn,nt,tn,tt}> Select GEMM datatype and layout to use for tests + --kernels=<{s|d|h|i|wmma_}gemm_{nn,nt,tn,tt}> Select GEMM datatype and layout to use for tests --peak= If true, only reports peak performance per kernel after profiling specified problem space. --save_workspace={*never,incorrect,always} Specifies when to save the GEMM inputs and results to the filesystem. --seed= Random seed used by the random number generator in initializing input matrices. diff --git a/tools/test/perf/testbench_options.h b/tools/test/perf/testbench_options.h index a58dedf1..dd267629 100644 --- a/tools/test/perf/testbench_options.h +++ b/tools/test/perf/testbench_options.h @@ -546,7 +546,7 @@ struct TestbenchOptions { << " --k=[:max depth[:step]] " << " Size of inner dimension of A and B. May specify a range with optional step size.\n" - << " --kernels=<{s|d|h|i|wmma}_gemm_{nn,nt,tn,tt}> " + << " --kernels={s|d|h|i|wmma_}gemm_{nn,nt,tn,tt} " << " Select GEMM datatype and layout to use for tests\n" << " --peak= "