Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
add cast storage benchmark params info
Browse files Browse the repository at this point in the history
  • Loading branch information
stefanhenneking committed Jul 20, 2017
1 parent dee3cbd commit d648709
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions benchmark/python/cast_storage.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,12 @@ def dns_to_csr(m, n, density, ctx, repeat):
check_call(_LIB.MXSetNumOMPThreads(ctypes.c_int(args.num_omp_threads)))

# params
# m number of rows
# n number of columns
# density density of the matrix
# num_repeat number of benchmark runs to average over
# contexts mx.cpu(), mx.gpu()
# note: benchmark different contexts separately; to benchmark cpu, compile without CUDA
m = [ 512, 512]
n = [50000, 100000]
density = [1.00, 0.80, 0.60, 0.40, 0.20, 0.10, 0.05, 0.02, 0.01]
Expand Down

0 comments on commit d648709

Please sign in to comment.