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

Commit

Permalink
Added indices, depth, ravel_data in default_params
Browse files Browse the repository at this point in the history
  • Loading branch information
connorgoggins committed Feb 27, 2020
1 parent 1af1d3d commit d91dca8
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions benchmark/opperf/rules/default_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -353,11 +353,16 @@
DEFAULT_Y = [(1024, 1024), (10000, 1), (10000, 100)]
DEFAULT_COND = [(1024,), (10000,), (10000,)]
DEFAULT_DEPTH = [0]

#DEFAULT_INDICES_LARGE_TENSOR = [(1, 2)]

# For ravel_multi_index op, ndim(shape) = 2; hence data NDArray's first dim = 2
# First dimension of input of ravel operator should match shape parameter dimension
# DEFAULT_SHAPE is reused for ravel_multi_index op
RAVEL_DATA = [(2, 1024)]

RAVEL_DATA_LARGE_TENSOR = [(2, 2**32)]

# For loss operators
DEFAULT_DATA_3d = [(1024, 100, 100)]
DEFAULT_LABEL = [(100,100)]
Expand Down Expand Up @@ -686,6 +691,9 @@
"eta": DEFAULT_ETA,
"eps": DEFAULT_EPSILON,
"stype": DEFAULT_STYPE,
"indices": DEFAULT_INDICES,
"depth": DEFAULT_DEPTH,
"ravel_data": RAVEL_DATA_LARGE_TENSOR,
"a": DEFAULT_A_LARGE_TENSOR,
"lhs_fill_element_0index": DEFAULT_LHS_FEI_LARGE_TENSOR,
"rhs_fill_element_0index": DEFAULT_RHS_FEI_LARGE_TENSOR,
Expand Down

0 comments on commit d91dca8

Please sign in to comment.